Add __all__ in the module