Add update logic to fab