Finish wiring up keyboard input for the new UI modes