修正Bug，优化代码