trying to fix the buttons