Tidied up the game over reset state