Passing remove callback into remove function