Add import: option to doctests