Add keyboard control