Refactor docstring