Update __init__.py

Fix any buggs