diff --git a/test/latex-test.md b/test/latex-test.md
new file mode 100644
index 0000000..68068e8
--- /dev/null
+++ b/test/latex-test.md
@@ -0,0 +1,8 @@
+# H1
+## H2
+### H3
+#### H4
+##### H5
+###### H6
+
+---
diff --git a/test/test-files/test-import.md b/test/test-files/test-import.md
index c4f0a5c..3c90501 100644
--- a/test/test-files/test-import.md
+++ b/test/test-files/test-import.md
@@ -1,7 +1,7 @@
 ```python {define=nop}
-appendChildren(element.content)
+return element.content
 ```
 
 ```python {define=opendatatask}
-println("Toto je praktická open-data úloha. V [odevzdávátku](https://ksp.mff.cuni.cz/h/odevzdavatko/) si necháte vygenerovat vstupy a odevzdáte příslušné výstupy. Záleží jen na vás, jak výstupy vyrobíte.")
+return import_md_list("Toto je praktická open-data úloha. V [odevzdávátku](https://ksp.mff.cuni.cz/h/odevzdavatko/) si necháte vygenerovat vstupy a odevzdáte příslušné výstupy. Záleží jen na vás, jak výstupy vyrobíte.")
 ```
diff --git a/test/test-files/test-partial.md b/test/test-files/test-partial.md
index 6450237..7c8b296 100644
--- a/test/test-files/test-partial.md
+++ b/test/test-files/test-partial.md
@@ -11,15 +11,13 @@ And things...
 
 :::
 
-``` {.python .run}
+```python {.run}
 # I set my own flags!
-ctx.set_flag("cat", True)
+context.set_flag("cat", True)
 ```
 
-``` {.python .run}
-println(f"The subdocument's title is \n\n# {ctx.get_metadata('title')}")
-println()
-println(f"The subdocument's subtitle is \n\n## {ctx.get_metadata('subtitle')}")
+```python {.run}
+return import_md_list(f"The subdocument's title is\n\n# {context.get_metadata('title')}\n\nThe subdocument's subtitle is\n\n## {context.get_metadata('subtitle')}")
 ```
 
 ```markdown {.group}
@@ -49,7 +47,7 @@ $$
 $$
 
 
-![This is a figure, go figure...](logo.svg){width=25%}
+![This is a figure, go figure...](logo.svg){width=25%}\
 
 ![This is a figure, go figure...](logo.pdf){width=50%}
 
diff --git a/test/test.md b/test/test.md
index 39b587e..f2b4e0f 100644
--- a/test/test.md
+++ b/test/test.md
@@ -3,10 +3,16 @@ title: 'Wooooo a title'
 subtitle: 'A subtitle'
 are_we_there_yet: False
 language: "en"
+lang: "en"
 ---
 [#test-files/test-import.md]{}
 
 # Hello world!
+## H2
+### H3
+#### H4
+##### H5
+###### H6
 
 This is an *example* **yay**!
 
@@ -25,14 +31,14 @@ This should only be shown to cats
 :::
 
 ```python {.run}
-ctx.set_flag("cat", True)
+context.set_flag("cat", True)
 ```
 
 ```python {.run}
-println(f"The main document's title is '{ctx.get_metadata('title')}'")
-ctx.set_metadata("a", {})
-ctx.set_metadata("a.b", {})
-ctx.set_metadata("a.b.c", "Bruh **bruh** bruh")
+context.set_metadata("a", {})
+context.set_metadata("a.b", {})
+context.set_metadata("a.b.c", "Bruh **bruh** bruh")
+return import_md_list(f"The main document's title is '{context.get_metadata('title')}'")
 ```
 
 ```python {style=native}
@@ -68,6 +74,8 @@ V pravém jízdním bruhu.
 
 [!nop]{a=b}<!-- A special command! WOW -->
 
+[Hehehehe čeština bružek]{lang=cs}
+
 > OOO a blockquote mate init
 >
 >> Nesting??
@@ -83,6 +91,8 @@ A link with the link in the link: <https://bruh.com>
 
 H~2~O is a liquid.  2^10^ is 1024.
 
+[Kuchařka](ksp:///kucharky/uvodni)
+
 [Underline]{.underline}
 
 :::{only=html}