update to models.py