models.py actualizado