#7 fix user input bug