Fixed a docstring and a comment