Update __init__.py

Fix for PR #5