Plotting functionality

Let's add some standard plotting functionality that is easy to call. These will therefore be the default plots that we include in the paper.