Modification for error handling (number entry solution)