Added some comments, and color option to draw function