[minor] Improved documentation based on code review.