Improve code for gameover screen. Still ugly visuals though.