Update __init__.py

I think..