Add key set to input handling