Add plot of energy.