优化代码结构，代码更漂亮

但是还需提升