Improved docstring documentation