Change getUserInput to take input from user