Add keyboard handling