Using a module for a water heater