Make the buttons work in design B