Add support for TE's energy system