diff --git a/README.md b/README.md
index ec03ce6..04ca5d9 100644
--- a/README.md
+++ b/README.md
@@ -20,9 +20,7 @@ We recommend to setup a virtual environment using Anaconda.
# Run the code
Our code supports training using GPUs or CPUs. It will prioritise GPUs if available (line 45 in main.py). You can also assign a particular card via CUDA_VISIBLE_DEVICES (e.g., the following commands use GPU card no.3).
-## Train Mouse2Vec Autoencoder
-
-
+## Train Mouse2Vec Autoencoder
Execute
```shell
CUDA_VISIBLE_DEVICES=3 python main.py --ssl True