Add new function to calculate the potential energy anomaly. Additional depth-averaging function included too.
Tidy up the function a bit to eliminate obsolete function to plot a model grid.