diff --git a/analyze.sh b/analyze.sh
index cdef7ae..7a189d4 100755
--- a/analyze.sh
+++ b/analyze.sh
@@ -1,6 +1,6 @@
 #!/usr/bin/bash
 
-find results ! -name 'train.txt' -type f -exec rm -f {} +
+rm results/*
 
 ./venv/bin/python3 distribution.py --graph --save | tee results/distribution.txt
 echo -e "\n\n\n\n"
@@ -17,3 +17,9 @@ echo -e "\n\n\n\n"
 ./venv/bin/python3 analyze_sleep.py --graph --save "results/Figure_18.png" | tee results/sleep.txt
 echo -e "\n\n\n\n"
 ./venv/bin/python3 analyze_absence.py --graph --save "results/Figure_19.png" | tee results/absence.txt
+echo -e "\n\n\n\n"
+./venv/bin/python3 train_nn.py --graph --save "results/Figure_20.png" | tee results/train.txt
+echo -e "\n\n\n\n"
+
+tar cvzf results.tar.gz results/
+zip results.zip results/*
diff --git a/results.tar.gz b/results.tar.gz
index 630d3e5..76c16a2 100644
Binary files a/results.tar.gz and b/results.tar.gz differ
diff --git a/results.zip b/results.zip
index f915ea7..a7ee28e 100644
Binary files a/results.zip and b/results.zip differ
diff --git a/train_nn.py b/train_nn.py
index 7c1d76a..12f55b2 100644
--- a/train_nn.py
+++ b/train_nn.py
@@ -11,8 +11,10 @@ parser = argparse.ArgumentParser(
     prog="train_nn"
 )
 parser.add_argument("-g", "--graph", action="store_true", default=False, help="Graph losses")
+parser.add_argument("-s", "--save", default="", help="Graph save location")
 args = parser.parse_args()
 graph = args.graph
+save = args.save
 
 
 class NeuralNetwork(nn.Module):
@@ -179,4 +181,7 @@ if graph:
 
     plt.legend()
     plt.tight_layout()
-    plt.show()
+    if save != "":
+        plt.savefig(save)
+    else:
+        plt.show()