Tweak docstring and comment.