From e7b964edc158d8e0862066e3eb02c18fed5e073d Mon Sep 17 00:00:00 2001
From: "Justin Terry (VM)" <juterry@microsoft.com>
Date: Fri, 16 Feb 2018 08:59:58 -0800
Subject: [PATCH] Updating 18.04 script prompts

---
 ubuntu/18.04/install.sh | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/ubuntu/18.04/install.sh b/ubuntu/18.04/install.sh
index 38b59f2..8576492 100644
--- a/ubuntu/18.04/install.sh
+++ b/ubuntu/18.04/install.sh
@@ -23,7 +23,12 @@ sudo apt-get update
 sudo apt install linux-image-4.14.0-17-generic
 
 if [ -f /var/run/reboot-required ]; then
-    reboot
+    echo
+    echo "A reboot is required in order to proceed with the install."
+    echo "Please reboot and re-run this script to finish the install."
+    echo
+
+    exit
 fi
 
 ###############################################################################
@@ -110,7 +115,7 @@ sudo systemctl start xrdp
 # Install Gmone Tweak
 sudo apt-get install gnome-tweak-tool -y
 
-#reboot
 echo
-echo "Reboot your machine to begin using XRDP"
+echo "Install is complete."
+echo "Reboot your machine to begin using XRDP."
 echo