Add key press logic