Extending update callback