Add new functions to __all__ in __init__.py