diff --git a/README.md b/README.md
index a183021..2b0b5f7 100644
--- a/README.md
+++ b/README.md
@@ -3,7 +3,9 @@
 Example usage:
 
 ```shell
-python main_cnn.py --debug train --dataset enwik9 --method optuna
+python main.py --debug train --dataset enwik9 --data-root ~/data/datasets/ml --method optuna --model transformer --model-save-path ~/data/ml-models/test-transformer.pt
+
+python benchmark.py --debug train --dataset enwik9 --data-root ~/data/datasets/ml --method optuna --model cnn --model-save-path ~/data/ml-models/test-cnn.pt
 ```
 
 ## Running locally