Added ControlState to track key input.