修改按钮添加判断