Add update function