changes to models.py