Docstrings updated