Adding key controlling feature