Buttons are working now