Implemented the potential energy budget equation
Fixed a few bugs, forgot to normalize the boussinesq term
Fixed a few bugs, the fluxes have to be computed with bz, whereas the
boussinesq term is calculated based on b
Added a stats script that plots the potential and kinetic energy budget
Updated the plotting script