diff --git a/merge-lbm-seq.R b/merge-lbm-seq.R
index 1ad11f6..5ec6991 100644
--- a/merge-lbm-seq.R
+++ b/merge-lbm-seq.R
@@ -3,18 +3,22 @@ library(stringr)
 args <- commandArgs(trailingOnly = TRUE)
 
 path <- args[1]
-mode <- args[2]
+
+
 
 base_folder <- str_remove(string = path, pattern = "/tmp[0-9]*$")
 print(base_folder)
 epoch <- str_extract(string = path, pattern = "(?<=tmp)([0-9]*)")
 
-flist <- list.files(path, pattern = paste0(mode, ".Rds"), full.names = TRUE)
+for (mode in c("seq", "para", "notrans")) {
+    flist <- list.files(path, pattern = paste0(mode, ".Rds"), full.names = TRUE)
+    if (length(flist) == 0) {
+        warning("No files corresponding to ", mode, " found.")
+        next
+    }
+    merged_res <- lapply(flist, readRDS)
 
-stopifnot("No files corresponding to mode found." = length(flist) > 0)
+    names(merged_res) <- paste0("Rank", seq_along(flist) + 1)
 
-merged_res <- lapply(flist, readRDS)
-
-names(merged_res) <- paste0("Rank", seq_along(flist) + 1)
-
-saveRDS(merged_res, file.path(base_folder, paste0(mode, "_", epoch, ".Rds")))
+    saveRDS(merged_res, file.path(base_folder, paste0(mode, "_", epoch, ".Rds")))
+}