still trying to fix the buttons