Add the __init__.py function for module import improvement