diff --git a/.gitignore b/.gitignore
index df0fd5f..065d54a 100644
--- a/.gitignore
+++ b/.gitignore
@@ -179,3 +179,6 @@ ipython_config.py
 # Remove previous ipynb_checkpoints
 #   git rm -r .ipynb_checkpoints/
 
+# runs folder
+baseline/runs/
+
diff --git a/README.md b/README.md
index ca27b31..52d565d 100644
--- a/README.md
+++ b/README.md
@@ -148,4 +148,4 @@ python evaluate.py
 | 125-dpd | 35.79 |
 | 250-dpd | 40.38 |
 
-> Note: The above results might change based on further experiments  
+> Note: The above preliminary results will change based on further experiments  
diff --git a/baseline/requirements.txt b/baseline/requirements.txt
index c074b70..19d6904 100644
--- a/baseline/requirements.txt
+++ b/baseline/requirements.txt
@@ -1,6 +1,6 @@
 numpy==1.17.4
 torch==1.6.0
 tqdm==4.38.0
-transformers==2.1.1
+transformers==2.2.2
 tensorboard==2.1.1
 tensorboardx==2.0
\ No newline at end of file
diff --git a/baseline/soloist_decode.py b/baseline/soloist_decode.py
index bb0a295..35449d4 100644
--- a/baseline/soloist_decode.py
+++ b/baseline/soloist_decode.py
@@ -9,10 +9,6 @@ import torch
 import torch.nn.functional as F
 import numpy as np
 
-import sys
-sys.path.append('.')
-sys.path.append('./transformers')
-sys.path.append('./transformers/')
 
 from transformers import GPT2Config
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
diff --git a/baseline/soloist_train.py b/baseline/soloist_train.py
index e391a08..14ccb44 100644
--- a/baseline/soloist_train.py
+++ b/baseline/soloist_train.py
@@ -11,11 +11,6 @@ import shutil
 import time
 import json
 
-import sys
-# sys.path.append('.')
-# sys.path.append('./transformers')
-# sys.path.append('./transformers/')
-
 import numpy as np
 import torch
 from torch.utils.data import DataLoader, Dataset, SequentialSampler, RandomSampler
diff --git a/baseline/train_baseline.sh b/baseline/train_baseline.sh
index 0f0b174..d2c50d8 100644
--- a/baseline/train_baseline.sh
+++ b/baseline/train_baseline.sh
@@ -34,7 +34,7 @@ SAVE_DIR="${SAVED_MODELS_BASELINE}"/"${experiment_folder}"
 mkdir -p "${SAVE_DIR}"
 
 echo "Trained Model checkpoints are saved in ${SAVE_DIR}"
-echo "Cleaning the directory (${experiment_folder}) to make sure it's empty!"
+echo "Cleaning the experiment directory (${experiment_folder}) to make sure it's empty!"
 
 # delete the contents of the dir (where model checkpoints will be saved!)
 rm -rf "${SAVE_DIR}"