diff --git a/src/common/budget.py b/src/common/budget.py
index f5379ac..3e4286d 100644
--- a/src/common/budget.py
+++ b/src/common/budget.py
@@ -1,6 +1,6 @@
 from dataclasses import dataclass
 
-from src.common.scheduler import Scheduler
+from assignment.simulator.src.scheduler import Scheduler
 
 @dataclass
 class Budget:
diff --git a/src/common/task.py b/src/common/task.py
new file mode 100644
index 0000000..93fe38e
--- /dev/null
+++ b/src/common/task.py
@@ -0,0 +1,11 @@
+from dataclasses import dataclass
+
+@dataclass
+class Task:
+    task_name:str
+    wcet:int
+    period:int
+    component_id:str
+    priority:int|None
+    
+