added menu options back in