修正代码完成，是代码结构更加清晰