Moved comment to docstring