* Add support for all state classes to energy cost sensor * Fix bug, adjust tests * Fix rebase mistake