@Torbjorn_Melbye that is a great suggestions, you are correct that we should add support for the heater as well. We have a feature freeze at the moment, but will add the heater class soon 
Also we will make it available in climate controls.
Thank you for contributing! 
