diff --git a/assemble-and-run.js b/assemble-and-run.js
index a8e6aea..45da349 100644
--- a/assemble-and-run.js
+++ b/assemble-and-run.js
@@ -11,14 +11,12 @@ const mode = process.argv[2];
 // console.log(`Reading ${filename}`);
 const filename = process.argv[3];
 const inputFile_str = fs.readFileSync(filename, 'utf8');
-let machineCode = assembler.assemble(inputFile_str);
-
-// console.group("Machine code output");
-// logMemory(machineCode);
-// console.groupEnd('Machine code output');
 
+let machineCode;
 if (mode === "debug") {
+  machineCode = assembler.assemble(inputFile_str, true);
   computer.runProgram(machineCode, true);
 } else {
+  machineCode = assembler.assemble(inputFile_str);
   computer.runProgram(machineCode, false);
 }
\ No newline at end of file