Adding model.py