Add lightsaber toggle function