diff --git a/rv32.sv b/rv32.sv
index 8f9df00..abf162d 100644
--- a/rv32.sv
+++ b/rv32.sv
@@ -18,10 +18,10 @@ module rv32 (
     rv32_fetch fetch (
         .clk(clk),
 
-        /* control in */
+        /* control in (from mem) */
         .branch_taken_in(mem_branch_taken),
 
-        /* data in */
+        /* data in (from mem) */
         .branch_pc_in(mem_branch_pc),
 
         /* data out */
diff --git a/rv32_decode.sv b/rv32_decode.sv
index 5ee51a9..54029fe 100644
--- a/rv32_decode.sv
+++ b/rv32_decode.sv
@@ -10,13 +10,15 @@
 module rv32_decode (
     input clk,
 
-    /* control in */
+    /* control in (from writeback) */
     input [4:0] rd_in,
     input rd_writeback_in,
 
     /* data in */
     input [31:0] pc_in,
     input [31:0] instr_in,
+
+    /* data in (from writeback) */
     input [31:0] rd_value_in,
 
     /* control out */
diff --git a/rv32_fetch.sv b/rv32_fetch.sv
index a6cb032..3813087 100644
--- a/rv32_fetch.sv
+++ b/rv32_fetch.sv
@@ -4,10 +4,10 @@
 module rv32_fetch (
     input clk,
 
-    /* control in */
+    /* control in (from mem) */
     input branch_taken_in,
 
-    /* data in */
+    /* data in (from mem) */
     input [31:0] branch_pc_in,
 
     /* data out */