tidy up get_user_choice function