Add puse button to key test