updated model.py