Problem 4 : Fixes user input message