Add documentation
Update docstrings to match numpydoc formatting