Improved prompt to allow for loops on invalid input