Update query.py

test