It is safe to add on extra socket by this way?
Original connection:
postimg.org/image/v6nshoygh/
Add-on connection:
postimg.org/image/vkoccsyu5/
The extra socket is for light usage only. Water heater is instant water heater without tank. The switch is double pole single switch.