diff --git a/code/train-model.sh b/code/train-model.sh
index b1dec86..a507bdd 100755
--- a/code/train-model.sh
+++ b/code/train-model.sh
@@ -2,7 +2,7 @@
 #
 #SBATCH --partition=longq    # Partition to submit to
 #
-#SBATCH --time=0-08:00         # Runtime in D-HH:MM
+#SBATCH --time=3-00:00         # Runtime in D-HH:MM
 #SBATCH --nodes=1
 #SBATCH --ntasks=1
 #SBATCH --ntasks-per-node=1
@@ -14,8 +14,9 @@ RUN_ID=$1
 MODEL_NAME=$2
 NUM_EPOCHS=$3
 NUM_USERS=$4
+TIME=$5
 
 source ~/mypython/bin/activate
 python main.py $RUN_ID $MODEL_NAME $NUM_EPOCHS $NUM_USERS
 
-
+# mailx -S smtp=mail.cs.umass.edu -a ../logs/usc-isi/${TIME}/train.log -a ../logs/usc-isi/${TIME}/train.err -s "Test mail" msurana@cs.umass.edu < /dev/null
diff --git a/code/training-job.sh b/code/training-job.sh
index 353d29e..89dcbb3 100755
--- a/code/training-job.sh
+++ b/code/training-job.sh
@@ -46,6 +46,9 @@ TIME=`(date +%Y-%m-%d-%H-%M-%S)`
                 -e ../logs/usc-isi/${TIME}/train.err \
                 -o ../logs/usc-isi/${TIME}/train.log \
                 --cpus-per-task $NUM_THREADS \
-                train-model.sh $RUN_ID $MODEL_NAME $NUM_EPOCHS $NUM_USERS
+                --mail-type=ALL \
+                --mail-user=mudit.bhargava90@gmail.com,mohit.surana95@gmail.com \
+                train-model.sh $RUN_ID $MODEL_NAME $NUM_EPOCHS $NUM_USERS $TIME
+
 exit