db.py updated