Fixed __init__.py and added __all__