Adding callback on remove method