Add the background potential energy to the plotting script