Redoing the documentation to do autodoc and adding necessary docstrings