Add doctests