A controller for an electric booster element in a water heater is described. The electric booster element is powered from mains power and the controller comprises a control module and a capacitive module adapted to store power and supply stored power to the control module. The control module produces a control signal for controlling a relay to supply or restrict mains power supply to said booster element, said control signal depending at least in part on time of use data.