docstring improvements