minor modification in __init__.py