修改代码结构

修改代码结构