Added __all__ string to __init__.py