Adds onUpdate callback