diff --git a/frontends/nextjs/src/components/get-component-icon.tsx b/frontends/nextjs/src/components/get-component-icon.tsx
index b90dbd8f7..f337d9c6b 100644
--- a/frontends/nextjs/src/components/get-component-icon.tsx
+++ b/frontends/nextjs/src/components/get-component-icon.tsx
@@ -4,31 +4,32 @@ import {
   AccountCircle,
   Article,
   Autorenew,
-  ChatBubbleOutline,
-  CheckBoxOutlined,
   CropFree,
-  CropSquare,
+  CropPortrait,
   FormatAlignLeft,
   GridView,
-  Layers,
+  Chat,
+  CheckBox,
+  LooksOne,
+  GridView,
   LocalOffer,
   Remove,
   TableChart,
   TextFields,
-  Title,
   ToggleOn,
   TouchApp,
   Tune,
   Verified,
+  ViewStream,
   ViewColumn,
   WarningAmber,
 } from '@mui/icons-material'
 
 const iconMap: Record<string, ComponentType<SvgIconProps>> = {
   Article,
-  Card: CropSquare,
-  Chat: ChatBubbleOutline,
-  CheckSquare: CheckBoxOutlined,
+  Card: CropPortrait,
+  Chat,
+  CheckSquare: CheckBox,
   CircleNotch: Autorenew,
   Columns: ViewColumn,
   CursorClick: TouchApp,
@@ -37,11 +38,11 @@ const iconMap: Record<string, ComponentType<SvgIconProps>> = {
   Minus: Remove,
   Seal: Verified,
   SlidersHorizontal: Tune,
-  Stack: Layers,
+  Stack: ViewStream,
   Table: TableChart,
   Tag: LocalOffer,
   TextAlignLeft: FormatAlignLeft,
-  TextHOne: Title,
+  TextHOne: LooksOne,
   TextT: TextFields,
   ToggleRight: ToggleOn,
   UserCircle: AccountCircle,
diff --git a/frontends/nextjs/src/lib/security/security-scanner.ts b/frontends/nextjs/src/lib/security/security-scanner.ts
index e012f40dc..8805468da 100644
--- a/frontends/nextjs/src/lib/security/security-scanner.ts
+++ b/frontends/nextjs/src/lib/security/security-scanner.ts
@@ -17,12 +17,11 @@ import { scanJavaScript } from './functions/scanners/scan-javascript'
 import { scanLua } from './functions/scanners/scan-lua'
 import { scanJSON } from './functions/scanners/scan-json'
 import { scanHTML } from './functions/scanners/scan-html'
+import { scanForVulnerabilities } from './functions/scanners/scan-for-vulnerabilities'
 import { sanitizeInput } from './functions/scanners/sanitize-input'
 import { getSeverityColor } from './functions/helpers/get-severity-color'
 import { getSeverityIcon } from './functions/helpers/get-severity-icon'
 
-import type { SecurityScanResult } from './functions/types'
-
 /**
  * SecurityScanner - Wrapper class for security scanning functions
  * 
@@ -36,6 +35,7 @@ export class SecurityScanner {
   scanJSON = scanJSON
   scanHTML = scanHTML
   sanitizeInput = sanitizeInput
+  scanForVulnerabilities = scanForVulnerabilities
 }
 
 // Default instance for convenience
@@ -50,38 +50,6 @@ export {
   scanLua,
   scanJSON,
   scanHTML,
+  scanForVulnerabilities,
   sanitizeInput
 }
-
-/**
- * Convenience function to scan code for vulnerabilities
- * Automatically detects type based on content or explicit type parameter
- */
-export function scanForVulnerabilities(
-  code: string, 
-  type?: 'javascript' | 'lua' | 'json' | 'html'
-): SecurityScanResult {
-  // Auto-detect type if not provided
-  if (!type) {
-    if (code.trim().startsWith('{') || code.trim().startsWith('[')) {
-      type = 'json'
-    } else if (code.includes('function') && code.includes('end')) {
-      type = 'lua'
-    } else if (code.includes('<') && code.includes('>')) {
-      type = 'html'
-    } else {
-      type = 'javascript'
-    }
-  }
-
-  switch (type) {
-    case 'lua':
-      return scanLua(code)
-    case 'json':
-      return scanJSON(code)
-    case 'html':
-      return scanHTML(code)
-    default:
-      return scanJavaScript(code)
-  }
-}