add doctest