Improved docstring
Improved docstring