Update models.py

wip