完善代码结构