Update models.py

correct one now?