简化代码，添加功能