Remove incorrect comment and add a docstring