Edited __init__.py