Fix previous commit by adding docstrings