diff --git a/network_attack_propagation.py b/network_attack_propagation.py
index 13bf9fe..658fee9 100644
--- a/network_attack_propagation.py
+++ b/network_attack_propagation.py
@@ -94,9 +94,12 @@ def update(num, layout, g_repr, ax, our_graph: Graph):
     if not our_graph.is_alive():
         return
 
-    our_graph.infect_step()
+    if num != 0:
+        our_graph.infect_step()
     ax.clear()
 
+    ax.set_title(f'Step: {num}', loc='right', fontsize=30)
+
     colors = ['red' if n.is_infected else 'blue' for n in g_repr]
     sizes = [200 if n.is_infected else 50 for n in g_repr]
     nx.draw(