重构代码 加入按钮