Actually, I must have been really tired when I made that diagram, because with the momentary switch directly connecting the batteries and the LED, the LED will only stay on while the button is being pressed. But if you were to put a momentary-switch Corbin LED driver in that circuit, it should work.