diff --git a/install.sh b/install.sh
index 1a9b8bd..6d00fd9 100755
--- a/install.sh
+++ b/install.sh
@@ -1,19 +1,30 @@
 #!/usr/bin/env sh
 
-#source /dev/stdin <<< "$(curl -sL http://go.scity.us/colorsinc)"
-#source /dev/stdin <<< "$(curl -sL http://go.scity.us/defaultinc)"
-
 DATA_DIR="/mnt/data"
-
+case "$(ubnt-device-info firmware || true)" in
+    1*)
+      DATA_DIR="/mnt/data"
+      ;;
+    2*)
+      DATA_DIR="/data"
+      ;;
+    3*)
+      DATA_DIR="/data"
+      ;;
+    *)
+      echo "ERROR: No persistent storage found." 1>&2
+      exit 1
+      ;;
+  esac
   
 
-echo -e "${idsCL[Green]}Installing UniFiOS-Utilities 'on-boot-script' system...${idsCL[Default]}"
+echo "Installing UniFiOS-Utilities 'on-boot-script' system..."
 echo
 # curl -fsL "https://raw.githubusercontent.com/unifi-utilities/unifios-utilities/HEAD/on-boot-script/remote_install.sh" | /bin/sh
 echo
-echo -e "${idsCL[Green]}Downloading hostname template file to: ${DATA_DIR}/on_boot.d/dns_custom_hostnames.sh${idsCL[Default]}"
+echo "${idsCL[Green]}Downloading hostname template file to: ${DATA_DIR}/on_boot.d/dns_custom_hostnames.sh"
 echo
-echo -e "${idsCL[LightCyan]}Edit the file above to add your custom DNS entires${idsCL[Default]}"
+echo "Edit the file above to add your custom DNS entires"