some changes in models.py