Added and tested the user input prompt