From 0aaade981da9af0cf1f90d762cdf32b87e49fab0 Mon Sep 17 00:00:00 2001
From: "Justin Terry (VM)" <juterry@microsoft.com>
Date: Wed, 14 Feb 2018 11:02:17 -0800
Subject: [PATCH] Cleaning up the install script to indicate states

---
 ubuntu/16.04/install.sh | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/ubuntu/16.04/install.sh b/ubuntu/16.04/install.sh
index 37211e5..88afcc9 100644
--- a/ubuntu/16.04/install.sh
+++ b/ubuntu/16.04/install.sh
@@ -21,7 +21,12 @@ sudo apt install -y git
 sudo apt install -y linux-azure
 
 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
 
 ###############################################################################
@@ -124,7 +129,6 @@ sudo make install
 # End XORGXRDP
 ###############################################################################
 
-#reboot
 echo
-echo "Reboot your machine to begin using XRDP"
-echo
\ No newline at end of file
+echo "Install is complete. Reboot your machine to begin using XRDP."
+echo