diff --git a/airbreaker.src b/airbreaker.src
index b96fbdd837f9da25b4c5cfe397e08dc62ccd972e..bf8ccdb10d7243484bace717e8a68a2615b4515f 100644
--- a/airbreaker.src
+++ b/airbreaker.src
@@ -85,14 +85,20 @@ network = networks[imports.utils.Choice(choice, "a wifi network")]
 
 bssid = network[0]
 required = network[1]
-if required < 0 then required ="*UNKNOWN*"
 essid = network[2]
 
-print("Once <color=white>"+required+"</color> ACKs have been generated, do Ctrl+C")
 // Nightly support
 if cryptools.hasIndex("aireplay") then
-	cryptools.aireplay(bssid,essid)
+	if required < 0 then
+		print("Please enter the number of ACKs you want to generate")
+		required = user_prompt().to_int
+		if typeof(required) == "string" or required < 0 then exit("Error: invalid number")
+	end if
+    print("Generating <color=white>"+required+"</color> ACKs, please wait")
+	cryptools.aireplay(bssid,essid,required)
 else
+    if required < 0 then required ="*UNKNOWN*"
+    print("Once <color=white>"+required+"</color> ACKs have been generated, do Ctrl+C")
 	args = "-b "+bssid+" -e "+essid
 	result = get_shell.launch(path, args)
 	if not result then
@@ -112,7 +118,7 @@ cryptools.airmon("stop", device)
 
 path = home_dir+"/file.cap"
 file = comp.File(path)
-if not file then import.utils.Exit("Couldn't find file, expected "+path)
+if not file then imports.utils.Exit("Couldn't find file, expected "+path)
 
 password = cryptools.aircrack(home_dir+"/file.cap")
 if not comp.connect_wifi(device, bssid, essid, password) then imports.utils.Exit("Couldn't connect !?")