Update models.py.