修改一下代码结构