Added support for doctests