优化了部分代码，修复一些小bug