AlMOST ADDED GAME OVER SCREEN! WOOP
Improving state handling when starting a game