Modified model.py