Update __init__.py

works with modules or class objects