Update __init__.py

Re-added the imports that were commented out by mistake