Add references to extra in setup.py and top-level __init__.py