添加函数优化代码