Implemented first version of background potential energy budget