Update __init__.py

Make it more visible to IDE's