diff --git a/AERO3220_HW4_block_diagram.md b/AERO3220_HW4_block_diagram.md
index 38ae464..70eae3a 100644
--- a/AERO3220_HW4_block_diagram.md
+++ b/AERO3220_HW4_block_diagram.md
@@ -24,15 +24,15 @@ flowchart TD
     O --> P[Run ode45 with eom_two_mass]
 
     P --> Q[Build Legend Strings]
-    Q --> R[Plot Figure 1: z1 vs time (5 cases)]
-    R --> S[Plot Figure 2: z2 vs time (5 cases)]
-    S --> T[Plot Figure 3: z1dot vs time (5 cases)]
-    T --> U[Plot Figure 4: z2dot vs time (5 cases)]
+    Q --> R[Plot Figure 1: z1 vs time for 5 cases]
+    R --> S[Plot Figure 2: z2 vs time for 5 cases]
+    S --> T[Plot Figure 3: z1dot vs time for 5 cases]
+    T --> U[Plot Figure 4: z2dot vs time for 5 cases]
     U --> V[Plot Figure 5: z1 and z2 together by case]
 
-    V --> W[Local Function: eom_two_mass(t,x,...)]
+    V --> W[Local Function eom_two_mass]
     W --> X[Unpack States: z1, z1dot, z2, z2dot]
-    X --> Y[Evaluate Input: u = Ffun(t)]
+    X --> Y[Evaluate Input u = Ffun of t]
     Y --> Z[Compute Accelerations: z1ddot and z2ddot]
     Z --> AA[Return State Derivative Vector dx with 4 elements]
     AA --> AB([End])
@@ -57,9 +57,9 @@ flowchart TD
     A([Start]) --> B[Define constants and masses: k1, k2, m1, m2]
     B --> C[Define simulation time: t0, tf, dt, tspan]
     C --> D[Set initial state vector x0 with 4 states]
-    D --> E[Define force inputs: F1=100, F2=100sin(0.25t), F3=100sin(0.50t)]
+    D --> E[Define force inputs F1 constant, F2 sine slow, F3 sine fast]
 
-    E --> F[Create case table: (Case, b, Force)]
+    E --> F[Create case table with case b and force]
     F --> G{i <= Number of Cases?}
 
     G -- Yes --> H[Load case i values: b_i, F_i]
@@ -82,7 +82,7 @@ flowchart TD
 ```mermaid
 flowchart LR
     A1[Inputs: t, x, m1, m2, k1, k2, b, Ffun] --> A2[Unpack states: z1, z1dot, z2, z2dot]
-    A2 --> A3[Compute input force: u = Ffun(t)]
+    A2 --> A3[Compute input force u from Ffun and t]
     A3 --> A4[Compute z1ddot from mass 1 equation]
     A4 --> A5[Compute z2ddot from mass 2 equation]
     A5 --> A6[Assemble derivative vector dx with 4 elements]