diff --git a/generator.php b/generator.php
index 0dd24ed..470c8e4 100644
--- a/generator.php
+++ b/generator.php
@@ -4,18 +4,25 @@ $base = file_get_contents(__DIR__."/src/base.sh");
 
 
 $dirs = [
-    __DIR__.'/src/functions/*.sh'
+    __DIR__.'/src/functions/*.sh',
+    __DIR__.'/src/ui/*.sh'
 ];
 
 $codes = [];
 
 foreach($dirs as $dir){
     foreach(glob($dir) as $file){
-        $codes[$file] = file_get_contents($file);
+        if(str_contains($dir, "functions")){
+            $codes['functions'][$file] = file_get_contents($file);
+        }
+        if(str_contains($dir, "ui")){
+            $codes['ui'][$file] = file_get_contents($file);
+        }
     }
 }
 
-$code = str_replace("{{ CODES }}", implode("\n", $codes), $base);
+$code = str_replace("{{ FUNCTIONS }}", implode("\n", $codes['functions']), $base);
+$code = str_replace("{{ UI }}", implode("\n", $codes['ui']), $code);
 
 try {
     $file = __DIR__.'/init.sh';