diff --git a/setup.ps1 b/setup.ps1
index 2e6e548..15669ab 100644
--- a/setup.ps1
+++ b/setup.ps1
@@ -65,7 +65,7 @@ do {
         Write-Host ''
         $answer = Read-Host 'Select number(s)'
         
-    } until ($answer -in @(1..5; 9))
+    } until ($answer -in @(1.2.3.4.5; 9))
 
     if ($answer -eq 9) { break }
 
@@ -118,8 +118,7 @@ do {
             )
             InstallTheApps $Apps
         }
-        '4' {
-            $apps = @( ## LAPTOP
+        '4' { $apps = @( ## LAPTOP
                 @{name = 'Intel.IntelDriverAndSupportAssistant' },
                 @{name = '9WZDNCRFJ4MV'; source = 'msstore' }, # Lenovo Vantage from MS Store
                 @{name = 'j5create.Driver-JUA365' } ## Not sure if this is correct yet!