Add key press event handleing