Fix breaking doctest

Modify the doc strings of the new added functions to
make doctests pass.