updating models.py
models.py