Completed all of the nice little docstrings.
Still need to make the plot prettier but it is in a working state at the moment.
Added an example image, updated the readme, update version #