Added a button to return to menu