Update models.py

mm