diff --git a/src/agent/task_types/construction_tasks.js b/src/agent/task_types/construction_tasks.js
index 4ad0c52..1ca100a 100644
--- a/src/agent/task_types/construction_tasks.js
+++ b/src/agent/task_types/construction_tasks.js
@@ -441,7 +441,7 @@ export function proceduralGeneration(m = 20,
             newZ >= 0 && newZ + newDepth <= p &&
             isSpaceValid(newX, newY, newZ, newLength, newWidth, newDepth)
         ) {
-            console.log(`Placing room at (${newX}, ${newY}, ${newZ}) with dimensions (${newLength}x${newWidth}x${newDepth})`);
+            // console.log(`Placing room at (${newX}, ${newY}, ${newZ}) with dimensions (${newLength}x${newWidth}x${newDepth})`);
             for (let di = 0; di < newDepth; di++) {
                 for (let dj = 0; dj < newLength; dj++) {
                     for (let dk = 0; dk < newWidth; dk++) {
diff --git a/tasks/construction_tasks/generate_multiagent_construction_tasks.js b/tasks/construction_tasks/generate_multiagent_construction_tasks.js
index 676a7fc..c34c1fb 100644
--- a/tasks/construction_tasks/generate_multiagent_construction_tasks.js
+++ b/tasks/construction_tasks/generate_multiagent_construction_tasks.js
@@ -148,8 +148,8 @@ function generateConstructionTasks(variants) {
 //Main: writes the generated tasks to a file.
 
 // VARIABLES TO CHANGE HERE
-const variants = 1
-const file = './test_multiagent_construction_tasks.json'
+const variants = 5
+const file = './train_multiagent_construction_tasks.json'
 
 const tasks = generateConstructionTasks(variants);
 // Clear existing file content