代码优化以及修改

对代码做了修改和优化