diff --git a/README.md b/README.md
index a95fa4c..dcaf386 100644
--- a/README.md
+++ b/README.md
@@ -29,7 +29,7 @@ cd CNS_drug_screening
 pytest
 ```
 
-### Run
+### Train the model
 ``` shell
 python ./train.py
 ```