Add some docstrings for the models