Documentation

[hammal]

We should perhaps think of a good way of adding some fundamental documentation.

Also the README.md file needs to be updated

[hammal]

For docstrings I suggest we follow this style guide

https://numpydoc.readthedocs.io/en/latest/format.html