Tidied up the codez so they can actually be used without exploding
Controller now tries to get input after drawing the screen