made changes to models.py