Still bug with game end when returning to mainMenu