fix docstrings and comments