Almost working with user input