对函数重新改组

根据新的需求修改了函数