docstring & comments fixed