Keyboard input is working, but incomplete