Add function to plot energetics data