Add update handler