diff --git a/lib/script_data.php b/lib/script_data.php
index eff7baf..5c2d1d1 100644
--- a/lib/script_data.php
+++ b/lib/script_data.php
@@ -1,17 +1,18 @@
 <?php
-function generateScriptData($phpArray) {
+function generateScriptData($phpArray):string {
     // Check if the array is associative and single-level
     if (is_array($phpArray) && count($phpArray) > 0 && count(array_filter(array_keys($phpArray), 'is_string')) === count($phpArray)) {
         // Generate JavaScript code to save each array element to local storage
-        echo "<script>";
+        $out = "<script>";
         foreach ($phpArray as $key => $value) {
             $escapedKey = addslashes($key); // Escape special characters in the key
             $escapedValue = addslashes($value); // Escape special characters in the value
 
-            echo "localStorage.setItem('$escapedKey', '$escapedValue');";
+            $out .= "localStorage.setItem('$escapedKey', '$escapedValue');";
         }
-        echo "</script>";
+        $out.= "</script>";
     } else {
-        echo "<script>console.error('Invalid PHP array. Must be single-level and associative.');</script>";
+        $out = "<script>console.error('Invalid PHP array. Must be single-level and associative.');</script>";
     }
+    return $out;
 }
\ No newline at end of file