Add useful imports to __init__.py