Yeah, that's exactly how I'm doing it, with a momentary switch it will have 4 "modes"
off, low, med, high, off low.....
With a clicky I would only code it with 3 modes, no "off" ..just click the switch to turn it off.
I've got a little setup here on the breadboard going with a micro...