Improve docstring documentation