Update query.py