updated docstrings / comments