modify model.py