Added callback to remove()