adding callback to remove