improved the docstrings