增加验证,优化代码