Small change to prompt_input