Smart water heaters