Add docstrings to models