优化代码，修复部分发现的问题