ENH: Add to __init__.py