Cleaned up docstrings in model.py