updated comments/docstrings