Yes that is the transformer.
Got it.. So basically from a connection standpoint I am removing the switch, hookup up the new switch and the adding a wire from the new switch to the neutral bus, and then same thing for ground. Which is what the drawing originally detailed.
Thank you.