Add restart/abort button & make 'try again' button go back to very start