整理代码结构

整理代码结构