对代码做了进一步优化