diff --git a/src/repertoire.r b/src/repertoire.r
index 86ca70e..ebfe7ce 100644
--- a/src/repertoire.r
+++ b/src/repertoire.r
@@ -1,13 +1,21 @@
 library(immuneSIM)
 
-sim_repertoire <- immuneSIM(
-  number_of_seqs = 1000,
-  species = "hs",
-  receptor = "tr",
-  chain = "b",
-  verbose = TRUE
-)
+generate_repertoires <- function(number_of_sequences) {
+  a_chain <- immuneSIM(
+    number_of_seqs = number_of_sequences,
+    species = "hs",
+    receptor = "tr",
+    chain = "a",
+    verbose = TRUE
+  )
+  b_chain <- immuneSIM(
+    number_of_seqs = number_of_sequences,
+    species = "hs",
+    receptor = "tr",
+    chain = "b",
+    verbose = TRUE
+  )
+  return(list("a_chain" = a_chain, "b_chain" = b_chain))
+}
 
-save(sim_repertoire, file = "data/repertoire")
 
-plot_report_repertoire(sim_repertoire, output_dir = "data/reports")
\ No newline at end of file