Thanks for all the helpful replies
I've now wired in the original water heater thermostat and it's working fine to automatically switch off the light at 40c (the upper limit), it is a bit slow at switching it back on as the temp needs to drop 6c before it switches it back on.
I'm trying to...