Passing remove callback intto remove function