Got to add some docstrings!! Ideally this should be setup to allow documentation to be generated by Sphinx