Added documentation to model.py