updated docstrings in functions and classes