diff --git a/models/v0.2.0/project.yml b/models/v0.2.0/project.yml
index c5eecd9..87b2cb5 100644
--- a/models/v0.2.0/project.yml
+++ b/models/v0.2.0/project.yml
@@ -39,8 +39,8 @@ description: |
   This step downloads all assets and prepares all the datasets and binaries for
   training use. For example, if you want to train `tl_calamancy_md`, run the following comand:
 
-  ```
-  bash scripts/tl_calamancy_md.sh
+  ```sh
+  MODEL=tl_calamancy_md scripts/train.sh
   ```
   
 
@@ -120,7 +120,7 @@ workflows:
   tl-calamancy-trf:
     - "train-parser-trf"
     - "train-ner-trf"
-    - "assemble"
+    - "assemble-trf"
 
 commands:
   - name: "setup-finetuning-data"
diff --git a/models/v0.2.0/scripts/tl_calamancy_md.sh b/models/v0.2.0/scripts/tl_calamancy_md.sh
deleted file mode 100755
index 08c918d..0000000
--- a/models/v0.2.0/scripts/tl_calamancy_md.sh
+++ /dev/null
@@ -1,3 +0,0 @@
-python -m spacy project run tl_calamancy . \
-    --vars.size md \
-    --vars.vectors vectors/floret-tl-md
\ No newline at end of file
diff --git a/models/v0.2.0/scripts/train.sh b/models/v0.2.0/scripts/train.sh
new file mode 100755
index 0000000..317c7c5
--- /dev/null
+++ b/models/v0.2.0/scripts/train.sh
@@ -0,0 +1,26 @@
+#!/bin/bash
+
+MODEL="tl_calamancy_md"  # Set this to tl_calamancy_md, tl_calamancy_lg, or tl_calamancy_trf as needed.
+
+case $MODEL in
+  tl_calamancy_md)
+    echo "Executing actions for tl_calamancy_md..."
+    python -m spacy project run tl-calamancy . \
+        --vars.size md \
+        --vars.vectors vectors/floret-tl-md
+    ;;
+  tl_calamancy_lg)
+    echo "Executing actions for tl_calamancy_lg..."
+    python -m spacy project run tl-calamancy . \
+        --vars.size lg \
+        --vars.vectors vectors/fasttext-tl
+    ;;
+  tl_calamancy_trf)
+    echo "Executing actions for tl_calamancy_trf..."
+    python -m spacy project run tl-calamancy-trf
+    # Add your commands here
+    ;;
+  *)
+    echo "Unknown MODEL: $MODEL"
+    ;;
+esac
\ No newline at end of file