Updated handle_input