That wiring diagram completely ignores the heating circuit. Keep in mind I'm not an electrician, but I would envision something like this:

Grounding, if any, would be taken care of through the device itself, switch always goes on the black wire.
Also, I would check the reviews of that heater you have in mind. Looks kind of spotty.