Simplified __init__.py to expose the appropriate API