fixed comments and docstrings. forgot on last commit