重构代码,增强功能！