Update query_db.py