Added changes in __init__.py