quedo algo extensa en mi solucion?

<canvas width="600" height="400"></canvas>

<script>
    var pantalla = document.querySelector('canvas');
    var pincel = pantalla.getContext('2d');

    pincel.fillStyle = 'grey';
    pincel.fillRect(0, 0, 600, 400);

    var puedoDibujar = false;
    var colorCirculo = ["red","green","blue"]
    var queColor = 0

    function dibujarCuadrado(x,y,color){
        pincel.fillStyle = color;
        pincel.fillRect(x,y,50,50);
    }
    dibujarCuadrado(0,0,"red")
    dibujarCuadrado(50,0,"green")
    dibujarCuadrado(100,0,"blue")
    function cambiarColor(evento){
    var x = evento.pageX - pantalla.offsetLeft;
    var y = evento.pageY - pantalla.offsetTop;
    if ((x > 0)&&
        (x < 51)&&
        (y > 0)&&
        (y < 51) ) {queColor = 0
    }

    if ((x > 51)&&
        (x < 101)&&
        (y > 0)&&
        (y < 51) ) {queColor = 1
    }

    if ((x > 101)&&
        (x < 151)&&
        (y > 0)&&
        (y < 51) ) {queColor = 2
    }
     return queColor
    }
    pantalla.onclick = cambiarColor

    function dibujarCirculo(evento) {

        if(puedoDibujar) {
            var x = evento.pageX - pantalla.offsetLeft;
            var y = evento.pageY - pantalla.offsetTop;
            pincel.fillStyle = colorCirculo[queColor];
            pincel.beginPath();
            pincel.arc(x, y, 5, 0, 2 * 3.14);
            pincel.fill();
        }

    }

    pantalla.onmousemove = dibujarCirculo;

    function habilitarDibujar() {

        puedoDibujar = true;
    }

    function deshabilitarDibujar() {

        puedoDibujar = false;
    }

    pantalla.onmousedown = habilitarDibujar;

    pantalla.onmouseup = deshabilitarDibujar;

</script>