my changes to main and the query.py