とりあえずB=BUTTON(2)の行を、WHILEループの中に入れれば良いんじゃないかな。