Modified to have the correct menu options