Create model.py