Add buttons back