diff --git a/update-dns.sh b/update-dns.sh
index 96c079c..cb779f8 100755
--- a/update-dns.sh
+++ b/update-dns.sh
@@ -21,7 +21,9 @@ else
 	echo -e "No network configs found\n"
 	exit 1
 fi
-echo 
+echo
+
+chmod 0600 ${netfile}
 
 read -e -p "Do you wish to continue changing the DNS? (Y/n) " choice
 case "$choice" in 
@@ -31,17 +33,19 @@ case "$choice" in
 	;;
 
 * ) if [ -f /etc/netplan/eth0.yaml ] || [ -f /etc/netplan/network.yaml ]; then
-		if [ "${1}" = "offsite" ]; then
+		# if [ "${1}" = "offsite" ]; then
 			sed -i 's/10.2.1.2,10.10.10.5/10.2.1.6,10.2.1.5/g' ${netfile}
-		else
+			sed -i 's/127.0.0.1,10.10.10.5/127.0.0.1,10.2.1.6g' ${netfile}
+		# else
 			sed -i 's/10.10.10.5,10.2.1.2/10.10.10.11,10.10.10.10/g' ${netfile}
-		fi
+			sed -i 's/127.0.0.1,10.10.10.10,10.2.1.2/127.0.0.1,10.10.10.10/g' ${netfile}
+			sed -i 's/127.0.0.1,10.10.10.11,10.2.1.2/127.0.0.1,10.10.10.11/g' ${netfile}
+		# fi
 
 		/usr/sbin/netplan apply
 		
-		cat ${netfile}
 	else
-		if [ "${1}" = "offsite" ]; then
+		# if [ "${1}" = "offsite" ]; then
 			sed -i 's/10.2.1.2 10.10.10.5/10.2.1.6 10.2.1.5/g' ${netfile}
 		else
 			sed -i 's/10.10.10.5 10.2.1.2/10.10.10.11 10.10.10.10/g' ${netfile}
@@ -49,8 +53,10 @@ case "$choice" in
 
 		/usr/sbin/service networking restart
 		
-		cat ${netfile}
 	fi
+	
+	cat ${netfile}
+	
 	echo -e "${Green}The DNS servers have been updated${Color_Off}"
 	;;
 esac