增加函数，提高代码简洁性

优化