modified 'models.py'