Adding doctest