models.py updated