Update models.py

changes to models