Update models.py

django is hard