Use doctest for documentation