Update models/db.py