Update model.py