Update input_query.py