Update __init__.py

Add missing imports