Import relevant stuff in __init__.py