Fixed and added docstrings for autodoc