Added docstrings to model.py