Addition of check and reprompt for user input