Moved code for drawing to another file to try to seperate the initialisation and drawing logic