Updated doc following code review.