Add better docstrings