Added missing input handler