Add a class for the heater