Use __all__ to expose public API