added callback for remove