Updating docstrings and adding tests
Adding documentation