fixed EnergyPlot.py main