STY: Improving docstring