Tweak docstring, comments.