bug fix; add buttons