diff --git a/src/main/java/com/kamco/cd/training/common/utils/HeaderUtil.java b/src/main/java/com/kamco/cd/training/common/utils/HeaderUtil.java
index f91bae9..0ed8c27 100644
--- a/src/main/java/com/kamco/cd/training/common/utils/HeaderUtil.java
+++ b/src/main/java/com/kamco/cd/training/common/utils/HeaderUtil.java
@@ -30,21 +30,15 @@ public final class HeaderUtil {
     RequestAttributes attrs = RequestContextHolder.getRequestAttributes();
 
     if (!(attrs instanceof ServletRequestAttributes servletAttrs)) {
-      log.info("!(attrs instanceof ServletRequestAttributes servletAttrs) : false");
       return false;
     }
 
     String acceptLanguage = servletAttrs.getRequest().getHeader("Accept-Language");
 
     if (acceptLanguage == null || acceptLanguage.isBlank()) {
-      log.info("!(acceptLanguage == null) : false");
       return false;
     }
 
-    log.info("acceptLanguage.toLowerCase(): {}", acceptLanguage.toLowerCase());
-    log.info(
-        "acceptLanguage.toLowerCase().startsWith(\"en\"): {}",
-        acceptLanguage.toLowerCase().startsWith("en"));
     return acceptLanguage.toLowerCase().startsWith("en");
   }
 }
diff --git a/src/main/java/com/kamco/cd/training/train/service/DockerTrainService.java b/src/main/java/com/kamco/cd/training/train/service/DockerTrainService.java
index 36d0931..ca3a799 100644
--- a/src/main/java/com/kamco/cd/training/train/service/DockerTrainService.java
+++ b/src/main/java/com/kamco/cd/training/train/service/DockerTrainService.java
@@ -515,7 +515,8 @@ public class DockerTrainService {
     }
 
     c.add("-v");
-    c.add(responseDir + ":/checkpoints");
+    // c.add(responseDir + ":/checkpoints");
+    c.add(responseDir + "/" + req.getOutputFolder() + ":/checkpoints");
 
     c.add("kamco-cd-train:latest");
 
@@ -523,7 +524,7 @@ public class DockerTrainService {
     c.add("/workspace/change-detection-code/run_evaluation_pipeline.py");
 
     addArg(c, "--dataset-folder", req.getDatasetFolder());
-    addArg(c, "--output-folder", req.getOutputFolder());
+    // addArg(c, "--output-folder", req.getOutputFolder());
 
     c.add("--epoch");
     c.add(modelFile);