Fixed some issues with gameover/win screen