Updated models in models.py