fix docstring and comment