edited model.py