Added user input for exercise 5