Update models.py

i'm a fucking awful programmer