Lighted switch like the Q8

hey guys, i have a modded Lantern that i would like to have a lighted switch similar to the Q8. It is a Rayovac sportmans extreme 3D. Currently the mods are XM-L2 4000K emitter with 1.52A Nanjg 101-AK-A1 driver 3 mode, and a regular reverse clicky switch and glow in the dark rubber boot. i’ve also soldered a male USB plug into the battery terminals so i could run the lantern on a power bank or cell phone wall charger. is there a switch i can just buy and replace or do i have to replace the driver as well? skill level is noob. i don’t know how to add/remove components of the driver. thanks.