Added input control