NEC code on hot water heater and disconnect. #2 G #168
This shows the complete procedure of how to convert an illegally installed water heater using romex to a completely up to code installation of a disconnect, flexible metallic conduit and THHN wire. NEC by the book methods and materials used. This also works for A/C units if the Neutrals are connected with a wire nut.