1 files changed, 8 insertions, 9 deletions
diff --git a/src/logic_unit.v b/src/logic_unit.v
index 0bbd642..f9f62a2 100644
--- a/src/logic_unit.v
+++ b/src/logic_unit.v
@@ -1,16 +1,15 @@
-module logic_unit #(
-  parameter N = 32
-)(
-  input [N-1:0] src0, src1,
-  input [1:0] op, // 00: AND, 01: OR, 10: XOR
-  output reg [N-1:0] result
+module logic_unit (
+  input [31:0] a, b,
+  input [1:0] op,
+  output reg [31:0] result
 );
   
 always @ (*) begin
   case (op)
-  2'b00: result <= src0 & src1;
-  2'b01: result <= src0 | src1;
-  2'b10: result <= src0 ^ src1;
+    2'b00: result <= a & b; // AND
+    2'b01: result <= a | b; // OR
+    2'b10: result <= a ^ b; // XOR
+    default: result <= 32'b0;
   endcase
 end