diff --git a/configs/callbacks/rank.yaml b/configs/callbacks/re-rank.yaml
similarity index 66%
rename from configs/callbacks/rank.yaml
rename to configs/callbacks/re-rank.yaml
index 64ff3e5..0c13ed9 100644
--- a/configs/callbacks/rank.yaml
+++ b/configs/callbacks/re-rank.yaml
@@ -1,5 +1,5 @@
 trainer:
   callbacks+:
-  - class_path: RankCallback
+  - class_path: ReRankCallback
     init_args:
       save_dir: ./runs
diff --git a/configs/data/run-dataset.yaml b/configs/data/run-dataset.yaml
new file mode 100644
index 0000000..4dd9344
--- /dev/null
+++ b/configs/data/run-dataset.yaml
@@ -0,0 +1,12 @@
+data:
+  class_path: LightningIRDataModule
+  init_args:
+    num_workers: 1
+    inference_batch_size: 4
+    inference_datasets:
+      - class_path: RunDataset
+        init_args:
+          run_path_or_id: null
+          depth: 100
+          sample_size: 100
+          sampling_strategy: top
\ No newline at end of file
diff --git a/configs/models/bi-encoder.yaml b/configs/models/bi-encoder.yaml
new file mode 100644
index 0000000..2afd575
--- /dev/null
+++ b/configs/models/bi-encoder.yaml
@@ -0,0 +1,4 @@
+model:
+  class_path: BiEncoderModule
+  init_args:
+    model_name_or_path: null
\ No newline at end of file
diff --git a/configs/models/colbert.yaml b/configs/models/colbert.yaml
new file mode 100644
index 0000000..37a4e49
--- /dev/null
+++ b/configs/models/colbert.yaml
@@ -0,0 +1,19 @@
+model:
+  class_path: BiEncoderModule
+  init_args:
+    model_name_or_path: ./tests/data/test_model
+    config:
+      class_path: ColConfig
+      init_args:
+        similarity_function: dot
+        query_aggregation_function:  sum
+        query_expansion: true
+        query_length: 32
+        doc_length: 256
+        normalize: true
+        add_marker_tokens: true
+        embedding_dim: 128
+        linear_bias: false
+    loss_functions:
+      - class_path: RankNet
+        
\ No newline at end of file
diff --git a/configs/models/cross-encoder.yaml b/configs/models/cross-encoder.yaml
new file mode 100644
index 0000000..9fa8bb5
--- /dev/null
+++ b/configs/models/cross-encoder.yaml
@@ -0,0 +1,4 @@
+model:
+  class_path: CrossEncoderModule
+  init_args:
+    model_name_or_path: null
\ No newline at end of file