diff --git a/baseline/decode_baseline.sh b/baseline/decode_baseline.sh
index 0ec7425..ebf9874 100644
--- a/baseline/decode_baseline.sh
+++ b/baseline/decode_baseline.sh
@@ -5,18 +5,18 @@
 # Check whether the required environment vars are set
 if [[ -z "${SAVED_MODELS_BASELINE}" ]]; then
     echo "SAVED_MODELS_BASELINE Environment variable not set. Run set_env_var.sh bash script"
-    return
+    exit 1
 fi
 
 # Check whether the MODEL_CHECKPOINT env var set
 if [[ -z "${MODEL_CHECKPOINT}" ]]; then
     echo "MODEL_CHECKPOINT Environment variable not set. Run \"export MODEL_CHECKPOINT=<path>\""
-    return
+    exit 1
 fi
 
 if [ ! -d "${SAVED_MODELS_BASELINE}/${MODEL_CHECKPOINT}" ]; then
     echo "Directory ${MODEL_CHECKPOINT} doesn't exist! Provide a valid Saved Model Checkpoint dir."
-    return
+    exit 1
 fi
 
 NS=5
diff --git a/baseline/requirements.txt b/baseline/requirements.txt
new file mode 100644
index 0000000..4944bae
--- /dev/null
+++ b/baseline/requirements.txt
@@ -0,0 +1,4 @@
+numpy==1.17.4
+torch==1.6.0
+tqdm==4.38.0
+transformers==2.1.1
\ No newline at end of file
diff --git a/baseline/soloist_train.py b/baseline/soloist_train.py
index 343c2cf..e391a08 100644
--- a/baseline/soloist_train.py
+++ b/baseline/soloist_train.py
@@ -12,9 +12,9 @@ import time
 import json
 
 import sys
-sys.path.append('.')
-sys.path.append('./transformers')
-sys.path.append('./transformers/')
+# sys.path.append('.')
+# sys.path.append('./transformers')
+# sys.path.append('./transformers/')
 
 import numpy as np
 import torch
diff --git a/baseline/train_baseline.sh b/baseline/train_baseline.sh
index b5ce0dd..6d2afa5 100644
--- a/baseline/train_baseline.sh
+++ b/baseline/train_baseline.sh
@@ -2,16 +2,29 @@
 # lr 1e-5 to 5e-5
 # mc_loss_efficient 0.1 to 1
 
-while getopts d: flag
+usage="$(basename "$0") [-d <data-split-name>]
+      Argument -d takes (few-shot) data split names.
+      Possible valid names : 50-dpd|100-dpd|125-dpd|250-dpd"
+
+while getopts :d: flag
 do
     case "${flag}" in
-        d) data_split=${OPTARG};;
+      d) data_split=${OPTARG};;
+      :) printf "missing argument for -%s\n" "$OPTARG" >&2; echo "$usage" >&2; exit 1;;
     esac
 done
 
+# check for mandatory/required -d argument
+# mandatory arguments
+if [ ! "$data_split" ]; then
+  echo "arguments -d must be provided"
+  echo "$usage" >&2; exit 1
+fi
+
 # Check whether the required environment vars are set
 if [[ -z "${SAVED_MODELS_BASELINE}" ]] || [[ -z "${PRE_TRAINED_SOLOIST}" ]]; then
     echo "Required Environment variables not set. First run set_env_var.sh"
+    exit 1
 fi
 
 datetime_now=$(date +"%Y%m%d")