修改代码结构
todo