Update doctest parts of comments