Add support for energy dispersion.