added support for on update callbacks