Added keyboard input to main