Fix all docstrings for models