Initial commit of model.py