fixing the buttons, I think