Added the option to go back to the main menu