add query.py