Add keypress handler