Add docstrings

[aburrell]

Update the code so that the comment blocks that contain document information are contained in docstrings: https://numpydoc.readthedocs.io/en/latest/format.html