Made a start on the drawing code