EaselJS Custom Display Object in ES6 - v.0.0.1

App.js

import MyButton from './MyButton.js';
const stage = new createjs.Stage("demo");

const btn = new MyButton('red');
stage.addChild(btn);
btn.x = 100;
btn.y = 100;
stage.update();

MyButton.js

// MyButton.js
export default class MyButton {

  constructor(color) {
    this.radius = 40;
    this.color = color || 'grey';
    return this.draw();
  }

  draw() {
    const circle = new createjs.Shape();
    circle.graphics.beginFill(this.color).drawCircle(0, 0, this.radius);
    return circle;
   }
}