Added wait for user input