改进代码结构, 为便于扩展做修改