Added keyboard control