Fixed a few comments and docstring