bug fixes and added energy dispersion in example plot;