Added the update callback