Important tweak to the above code - use keydown, not keypress!