add doctest
doctest