整理代码

重新设计代码结构