updated doctest