Add doctest