Documentation

Adding docstrings