优化部分代码

改善结构