代码优化

对代码进行了一些优化