Update __init__.py

bug fixes