add the model.py