fixing model.py