Fixed Exercise 5, did not ask user for input