修改代码部分结构，进行了重构