Add docstrings to model methods and functions