update after comments on pr