Updated the get_input() function