改进了代码, 做出了一些小修正