Add __init__.py for package