Tweak docstring