diff --git a/examples/di/atomization_capacity_plan.py b/examples/di/atomization_capacity_plan.py
index c84a2376b..eb86f9bd2 100644
--- a/examples/di/atomization_capacity_plan.py
+++ b/examples/di/atomization_capacity_plan.py
@@ -1,41 +1,30 @@
+import fire
+
 from metagpt.logs import logger
 from metagpt.roles.di.team_leader import TeamLeader
-import fire
 
 
 async def main():
     tl = TeamLeader()
     logger.info("\n=== Adding Initial Tasks ===")
     tl.planner.plan.append_task(
-        task_id="T1",
-        dependent_task_ids=[],
-        instruction="Create Product Requirements Document (PRD)",
-        assignee="Alice"
+        task_id="T1", dependent_task_ids=[], instruction="Create Product Requirements Document (PRD)", assignee="Alice"
     )
     tl.planner.plan.append_task(
-        task_id="T2",
-        dependent_task_ids=["T1"],
-        instruction="Design System Architecture",
-        assignee="Bob"
+        task_id="T2", dependent_task_ids=["T1"], instruction="Design System Architecture", assignee="Bob"
     )
 
-    # 2. Add Development Tasks
+    # 1. Add Development Tasks
     logger.info("\n=== Adding Development Tasks ===")
     tl.planner.plan.append_task(
-        task_id="T3",
-        dependent_task_ids=["T2"],
-        instruction="Implement Core Function Modules",
-        assignee="Alex"
+        task_id="T3", dependent_task_ids=["T2"], instruction="Implement Core Function Modules", assignee="Alex"
     )
 
     tl.planner.plan.append_task(
-        task_id="T4",
-        dependent_task_ids=["T2"],
-        instruction="Implement User Interface",
-        assignee="Alex"
+        task_id="T4", dependent_task_ids=["T2"], instruction="Implement User Interface", assignee="Alex"
     )
 
-    # 3. Complete Some Tasks
+    # 2. Complete Some Tasks
     logger.info("\n=== Execute and Complete Tasks ===")
     logger.info(f"Current Task: {tl.planner.plan.current_task.instruction}")
     tl.planner.plan.finish_current_task()  # Complete T1
@@ -43,25 +32,22 @@ async def main():
     logger.info(f"Current Task: {tl.planner.plan.current_task.instruction}")
     tl.planner.plan.finish_current_task()  # Complete T2
 
-    # 4. Replace Tasks
+    # 3. Replace Tasks
     logger.info("\n=== Replace Task ===")
     tl.planner.plan.replace_task(
         task_id="T3",
         new_dependent_task_ids=["T2"],
         new_instruction="Implement Core Function Modules (Add New Features)",
-        new_assignee="Senior_Developer"
+        new_assignee="Senior_Developer",
     )
 
-    # 5. Add Testing Tasks
+    # 4. Add Testing Tasks
     logger.info("\n=== Add Testing Tasks ===")
     tl.planner.plan.append_task(
-        task_id="T5",
-        dependent_task_ids=["T3", "T4"],
-        instruction="Execute Integration Tests",
-        assignee="Edward"
+        task_id="T5", dependent_task_ids=["T3", "T4"], instruction="Execute Integration Tests", assignee="Edward"
     )
 
-    # 6. Reset Task Demonstration
+    # 5. Reset Task Demonstration
     logger.info("\n=== Reset Task ===")
     logger.info("Reset Task T3 (This will also reset T5 which depends on it)")
     tl.planner.plan.reset_task("T3")
@@ -74,5 +60,6 @@ async def main():
     for task in tl.planner.plan.tasks:
         logger.info(f"- {task.task_id}: {task.instruction} (Completed: {task.is_finished})")
 
+
 if __name__ == "__main__":
     fire.Fire(main)