Make it plot energy!