Botones para crear dibujos

[djplana]

Hola, soy nuevo por aquí, estoy aprendiendo mediante unos libros y me gustaría hacer lo que me pide en las sugerencias.

Necesitaría el código ActionScript 3.0 para flash cs5 para que un botón (boton_cuadrados) dibuje en forma de cuadrados y otro botón (boton_circulos) dibuje en forma de círculos.

Actualmente el código con el que acaba el tema en el libro, es para que pulsando E en el teclado dibuje círculos y pulsando R dibuje cuadrados, ahora no consigo cambiarlo para que sea pulsando botones.

Dejo el código actual para que se hagan una idea, espero que alguien me diga como es, un saludo y gracias de antemano.

var shapeType:String = "ellipse";
var color:Number;

stage.addEventListener(MouseEvent.MOUSE_DOWN, startDrawing);
stage.addEventListener(MouseEvent.MOUSE_UP, stopDrawing);
stage.addEventListener(KeyboardEvent.KEY_DOWN, keyDownHandler );

function keyDownHandler(e:KeyboardEvent):void {
if (e.keyCode == "E".charCodeAt()) {
shapeType = "ellipse";
} else if (e.keyCode == "R".charCodeAt()) {
shapeType = "rectangle";
}
}
function startDrawing(e:MouseEvent):void {
stage.addEventListener(MouseEvent.MOUSE_MOVE, makeShapes);
color = Math.random() * 0xFFFFFF;
}

function stopDrawing(e:MouseEvent):void {
stage.removeEventListener(MouseEvent.MOUSE_MOVE, makeShapes)
}

function makeShapes(e:MouseEvent):void {

if(shapeType == "ellipse") {
var ellipse:Ellipse = new Ellipse(10, 10, color);
addChild(ellipse);
ellipse.x = mouseX;
ellipse.y = mouseY;
}else if (shapeType == "rectangle") {
var rectangle:Rect = new Rect(5, 15, color);
addChild(rectangle);
rectangle.x = mouseX;
rectangle.y = mouseY;
}
}