Updated the energy checker python script to save the data and figure