Add keyboard input handling