Update model.py

small change