Update labsecurity

Author: dylanmeca

labsecurity

@@ -42,25 +42,29 @@ def terminal ():
                                global ip1
                                print (Style.BRIGHT + Fore.GREEN + "ip =>", prompt[2])
                                ip1 = prompt[2]
-                         if prompt[1] == 'port':
+                         elif prompt[1] == 'port':
                                global port1
                                print (Style.BRIGHT + Fore.GREEN + "port =>", prompt[2])
                                port1 = prompt[2]
+                         else:
+                               print (Style.BRIGHT + Fore.RED + "[*] ERROR, use the help command for more information")
                    elif prompt[0] == 'use':
                          global use1
                          #Here is the part where one of the following categories is used
                          if prompt[1] == 'scanweb':
                                print (Style.BRIGHT + Fore.GREEN + "use =>", prompt[1])
                                use1 = prompt[1]
-                         if prompt[1] == 'scanports':
+                         elif prompt[1] == 'scanports':
                                print (Style.BRIGHT + Fore.GREEN + "use =>", prompt [1])
                                use1 = prompt[1]
-                         if prompt[1] == 'scanport':
+                         elif prompt[1] == 'scanport':
                                print (Style.BRIGHT + Fore.GREEN + "use =>", prompt[1])
                                use1 = prompt[1]
-                         if prompt[1] == 'scanip':
+                         elif prompt[1] == 'scanip':
                                print (Style.BRIGHT + Fore.GREEN + "use =>", prompt[1])
                                use1 = prompt[1]
+                         else:
+                               print (Style.BRIGHT + Fore.RED + "[*] ERROR, use the help command for more information")
                    elif prompt[0] == 'clear':
                          os.system ("clear")
                    elif prompt[0] == 'exit' or prompt[0] == 'quit':
@@ -69,15 +73,17 @@ def terminal ():
                          if use1 == 'scanweb':
                              print (Style.BRIGHT  + Fore.BLUE + "[*] Scanning website...")
                              scanner.scanweb (ip1)
-                         if use1 == 'scanports':
+                         elif use1 == 'scanports':
                              print (Style.BRIGHT + Fore.BLUE + "[*] Scanning ports...")
                              scanner.scanports(ip1)
-                         if use1 == 'scanport':
+                         elif use1 == 'scanport':
                              print (Style.BRIGHT + Fore.BLUE + "[*] Scanning port...")
                              scanner.scanport (ip1, port1)
-                         if use1 == 'scanip':
+                         elif use1 == 'scanip':
                              print (Style.BRIGHT + Fore.BLUE + "[*] Scanning ip...")
                              scanner.scanip (ip1)
+                         else:
+                             print (Style.BRIGHT + Fore.RED + "[*] ERROR, use the help command for more information")
                    else:
                          print (Style.BRIGHT + Fore.RED + "[*] ERROR, use the help command for more information")