Refactor menu options and add prompt for checkout