Update models.py

db