Moved EnergyTypes into a inner class and update SetEnergy