Added doctest for #159