Add support for notify on update