Moved a comment to docstring