improve docstrings