Fixed the get_input function.