diff --git a/network_attack_propagation.py b/network_attack_propagation.py
new file mode 100644
index 0000000..bcabc56
--- /dev/null
+++ b/network_attack_propagation.py
@@ -0,0 +1,18 @@
+class Node:
+    def __init__(self):
+        self.is_infected = False
+
+
+class Edge:
+    def __init__(self, node_a: Node, node_b: Node, weight: float):
+        self.node_a = node_a
+        self.node_b = node_b
+        self.weight = weight
+
+
+class Graph:
+    def __init__(self):
+        self.edges = []
+
+    def add_edge(self, edge: Edge):
+        self.edges.append(edge)
\ No newline at end of file