优化了一下代码