updated query.py