better docstring
improvements