Add in an option to quit;