修改bug优化代码