still working on the buttons