修复bug，优化代码