Add the APE budget to the plotting script of drycbl