Add doctest's $setup section