Continue work on the GUI - added code for drawing