Changed comments and docstring