Update doc following code review.