diff --git a/src/test/kotlin/ParserTest.kt b/src/test/kotlin/ParserTest.kt
index 66067c4..b922dd0 100644
--- a/src/test/kotlin/ParserTest.kt
+++ b/src/test/kotlin/ParserTest.kt
@@ -1,7 +1,9 @@
 package main
 
 import junit.framework.TestCase.assertEquals
+import junit.framework.TestCase.assertTrue
 import org.junit.Test
+import java.io.File
 
 class ParserTest {
 
@@ -14,16 +16,43 @@ class ParserTest {
 
     @Test
     fun testTakeOnlyJs() {
-        val fileToTest = "src/test/kotlin/HTMLFileWithScript.html"
-        val expectedJsCode = """
-        function greet() {
-            return "Hello from JavaScript!";
-        }
-        console.log(greet());""".trimIndent().replace("\\s+".toRegex(), " ")
-
+        val fileToTest = "src/test/resources/FileWithScript.html"
+        val expectedJsCode = File("src/test/resources/OnlyTheScript.txt").readText().trimIndent().replace("\\s+".toRegex(), " ")
         val output = takeOnlyJs(fileToTest).trimIndent().replace("\\s+".toRegex(), " ")
         assertEquals(expectedJsCode, output)
     }
 
+    @Test
+    fun testJsWords() {
+        // Assume you have a sample code map
+        val codeMap = mapOf("word1" to 2, "word2" to 3, "javascript" to 5)
+        val output = jsWords(codeMap)
+        // Assert the expected output based on the filtering logic
+        assertEquals(mapOf("word1" to 2, "word2" to 3, "javascript" to 5), output)
+    }
+
+    @Test
+    fun testSplitLanguages() {
+        // Assume you have a sample HTML file with <script> and <style> tags
+        val fileWithHTMLCSSJS = "src/test/resources/FileWithStyleAndScript.html"
+        val expectedJsCode = "function greet() {\n    return \"Hello from JavaScript!\";\n}\n\nconsole.log(greet());".trimIndent().replace("\\s+".toRegex(), " ")
+        val expectedCssCode = "body {\n    background-color: #f0f0f0;\n}".trimIndent().replace("\\s+".toRegex(), " ")
+
+        // Call the parser function with the sample HTML file
+        splitLanguages(fileWithHTMLCSSJS)
+
+        // Verify the content of the generated files
+        val actualJsCode = File("javascript.js").readText().trimIndent().replace("\\s+".toRegex(), " ")
+        val actualHtmlCode = File("html.html").readText()
+        val actualCssCode = File("css.css").readText().trimIndent().replace("\\s+".toRegex(), " ")
+
+        // Assertions
+        assertEquals(expectedJsCode, actualJsCode)
+        assertEquals(expectedCssCode, actualCssCode)
+
+        // Ensure that HTML content is modified as expected
+        assertTrue(actualHtmlCode.contains("<script src=\"javascript.js\"></script>"))
+        assertTrue(actualHtmlCode.contains("<link href=\"css.css\" rel=\"stylesheet\" />"))
+    }
 
 }
diff --git a/src/test/kotlin/HTMLFileWithScript.html b/src/test/resources/FileWithScript.html
similarity index 100%
rename from src/test/kotlin/HTMLFileWithScript.html
rename to src/test/resources/FileWithScript.html
diff --git a/src/test/resources/FileWithStyleAndScript.html b/src/test/resources/FileWithStyleAndScript.html
new file mode 100644
index 0000000..46f6393
--- /dev/null
+++ b/src/test/resources/FileWithStyleAndScript.html
@@ -0,0 +1,23 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Sample HTML with Script and Style</title>
+    <style>
+        body {
+            background-color: #f0f0f0;
+        }
+    </style>
+</head>
+<body>
+<h1>Hello, World!</h1>
+<script>
+    function greet() {
+        return "Hello from JavaScript!";
+    }
+
+    console.log(greet());
+</script>
+</body>
+</html>
\ No newline at end of file
diff --git a/src/test/kotlin/OnlyTheScript.txt b/src/test/resources/OnlyTheScript.txt
similarity index 72%
rename from src/test/kotlin/OnlyTheScript.txt
rename to src/test/resources/OnlyTheScript.txt
index 9076a20..eb8fdd7 100644
--- a/src/test/kotlin/OnlyTheScript.txt
+++ b/src/test/resources/OnlyTheScript.txt
@@ -1,3 +1,4 @@
+
 function greet() {
     return "Hello from JavaScript!";
 }
