Added import in comp.py doctests