just getting around to fixing my headlights and all this other sh*t for the bike but was thinking would this work?

recently saw someone who had the same problem with their headlights and they just wired in their own switch powered from the battery to work em

i've bought myself a toggle switch which has 3 clicks and was thinking this is how i would wire it to run the lights:



1 in the diagram would be off
2 in the diagram would be low beams
3 in the diagram would be high beams

was thinking of sourcing the light plugs to mod and then splicing into the green wire for my power supply... but im no beginner in electronics and was wondering what everyone elses thought on this would be.