Small update on the docstrings