Added 2 buttons to key handler