diff --git a/runme.sh b/runme.sh
index c0845eec..88c9e7a3 100755
--- a/runme.sh
+++ b/runme.sh
@@ -13,19 +13,20 @@ fi
 
 #Arguments
 
+DISTRO=$(lsb_release -i -s)
 
-while getopts ud:r: option
+while getopts ul:d:r: option
 do
     case "${option}" in
         u) UNATTEND="true" && echo "UNATTEND";;
+        l) DISTRO=${OPTARG} && echo $DISTRO;;
         d) DESKTOP=${OPTARG} && echo $DESKTOP;;
         r) REPO=${OPTARG} && echo $REPO;;
     esac
 done
 
-DISTRO=$(lsb_release -i -s)
 
-if [ $UNATTEND != "true" ];
+if [[ $UNATTEND != "true" ]];
 then
     whiptail --backtitle "Pentest Build Script" --title "Welcome" --msgbox "This script will help you build a custom pentesting distribution." 7 70
 
@@ -104,7 +105,7 @@ Parrot)
 ;;
 esac
 
-if [ $UNATTEND != "true" ];
+if [[ $UNATTEND != "true" ]];
 then
 whiptail --backtitle "Pentest Build Script" --title "Finished" --msgbox "Fingers crossed there were no errors... Your image should be done!" 7 70
 fi