改进了代码结构