改善代码结构