Update models.py

this might break everything and I blame @individuo7