edit __init__.py