diff --git a/setup.py b/setup.py
index b3d4b28c..33d4bf5e 100644
--- a/setup.py
+++ b/setup.py
@@ -76,7 +76,7 @@ def revert():
             "Operating System :: OS Independent",
         ],
         package_data={
-            "taskweaver": ["**/*.yaml"],
+            "taskweaver": ["**/*.yaml", "**/*.yml"],
         },
         entry_points={
             "console_scripts": ["taskweaver=taskweaver.__main__:main"],
diff --git a/taskweaver/ces/environment.py b/taskweaver/ces/environment.py
index 8984e69f..9f0eb147 100644
--- a/taskweaver/ces/environment.py
+++ b/taskweaver/ces/environment.py
@@ -259,15 +259,15 @@ def start_session(
             )
 
             tick = 0
-            while tick < 10:
+            while tick < 30:
                 container.reload()
                 if container.status == "running" and os.path.isfile(connection_file):
                     logger.info("Container is running and connection file is ready.")
                     break
                 time.sleep(1)  # wait for 1 second before checking again
                 tick += 1
-            if tick == 10:
-                raise Exception("Container is not ready after 10 seconds")
+            if tick == 30:
+                raise Exception("Container is not ready after 30 seconds")
 
             # save the ports to ces session dir
             port_bindings = container.attrs["NetworkSettings"]["Ports"]
diff --git a/taskweaver/ces/kernel/__init__.py b/taskweaver/ces/kernel/__init__.py
new file mode 100644
index 00000000..e69de29b
diff --git a/taskweaver/ces/manager/__init__.py b/taskweaver/ces/manager/__init__.py
new file mode 100644
index 00000000..e69de29b
diff --git a/taskweaver/ces/runtime/__init__.py b/taskweaver/ces/runtime/__init__.py
new file mode 100644
index 00000000..e69de29b
diff --git a/taskweaver/chat/__init__.py b/taskweaver/chat/__init__.py
new file mode 100644
index 00000000..e69de29b
diff --git a/taskweaver/ext_role/echo/__init__.py b/taskweaver/ext_role/echo/__init__.py
new file mode 100644
index 00000000..e69de29b
diff --git a/taskweaver/ext_role/web_explorer/__init__.py b/taskweaver/ext_role/web_explorer/__init__.py
new file mode 100644
index 00000000..e69de29b
diff --git a/taskweaver/ext_role/web_search/__init__.py b/taskweaver/ext_role/web_search/__init__.py
new file mode 100644
index 00000000..e69de29b
diff --git a/taskweaver/module/__init__.py b/taskweaver/module/__init__.py
new file mode 100644
index 00000000..e69de29b