task1/query.py seems to work