diff --git a/src/lambo/examples/run_training.py b/src/lambo/examples/run_training.py
index 34c8b889b55829764a7d0710172e31849e7b3035..95544040b5109af203e2e05b3786f7459487f935 100644
--- a/src/lambo/examples/run_training.py
+++ b/src/lambo/examples/run_training.py
@@ -17,9 +17,11 @@ if __name__=='__main__':
     
     # Read available languages
     languages_file_str = resources.read_text('lambo.resources', 'languages.txt', encoding='utf-8', errors='strict')
-    languages = [line.split(' ')[0] for line in languages_file_str.split('\n')]
+    languages = [line.split(' ')[0] for line in languages_file_str.split('\n') if not line[0] == '#']
     
     for i in range(len(languages)):
+        if len(sys.argv)>3 and i % 5 != int(sys.argv[3]):
+            continue
         language = languages[i]
         if (outpath / (language + '.pth')).exists():
             continue
diff --git a/src/lambo/examples/run_training_pretrained.py b/src/lambo/examples/run_training_pretrained.py
index 0e7f7ee88cdbf2851aeab3a6ff7579efa2666666..6a81eb240a76e4e4389af7e36634f583b894c12c 100644
--- a/src/lambo/examples/run_training_pretrained.py
+++ b/src/lambo/examples/run_training_pretrained.py
@@ -20,7 +20,7 @@ if __name__=='__main__':
     lines = [line.strip() for line in languages_file_str.split('\n') if not line[0] == '#']
     
     for i, line in enumerate(lines):
-        if i % 5 != int(sys.argv[4]):
+        if len(sys.argv)>4 and i % 5 != int(sys.argv[4]):
             continue
         parts = line.split()
         model = parts[0]