Add input_with_timeout