Fixed/added some comments/docstrings