Made the ability to go back to the main menu when the game is done