增加cleanup函数