Add Energy class.
Untested.