Added ability to go back to main menu after game over