Improve docstrings