Add docstrings for simulation models