Make main draw function into switch