优化代码,修改BUG