Added the option to go back to the main menu from the stepping stone automata