From b813aaeae4da2c83c7fa09e72314f8a5c5a24f9b Mon Sep 17 00:00:00 2001
From: David Schroeder <david@schroedercity.com>
Date: Sun, 4 May 2025 10:05:08 -0500
Subject: [PATCH] Update storage-test.sh

---
 storage-test.sh | 22 ++++++++++++----------
 1 file changed, 12 insertions(+), 10 deletions(-)

diff --git a/storage-test.sh b/storage-test.sh
index fc836a5..cf1d5b0 100755
--- a/storage-test.sh
+++ b/storage-test.sh
@@ -104,16 +104,18 @@ IO4kRandWrite=0
 # Did the user give a folder ?
 ChosenPartition=""
 if [ "$1" = "" ]; then
-  df=0; declare -A driveSel
-  echo "$(/bin/df -kTh | grep ext4)" | while IFS= read -r line ; do
-	  ((df++))
-	  path=$(echo $line | cut -d'%' -f2)
-	  driveSel["${df}"] = "${path// }"
-	  echo "$df) ${path// }"
-  done
-  
-  exit 0
-  cd /
+	echo
+	Print_Style "Select a partition to test from:" "$YELLOW"
+	df=0; declare -A driveSel
+	echo "$(/bin/df -kTh | grep ext4)" | while IFS= read -r line ; do
+		((df++))
+		path=$(echo $line | cut -d'%' -f2)
+		driveSel["${df}"] = "${path// }"
+		Print_Style "$df) ${path// }" "$BLUE"
+	done
+
+	exit 0
+	cd \
 else
   if [ ! -d "$1" ]; then
     Print_Style "Your chosen partition (folder) does not exist! Provide a good one or run without parameters to check the rootfs" "$RED"