__init__.py update v2.0