Moved the drawing code into own visualiser class