added documentation to model.py