Add support for energy plotting and multi data visualization