updated db.py