code cleaning
- fix the canvas size
- fix #9