Update models.py

not sure if this actually works