updates to models.py