Adding keyboard control