Add buttons back to ToolSwitch