diff --git a/src/psm_parser.py b/src/psm_parser.py
index ebc0f41..90507d8 100644
--- a/src/psm_parser.py
+++ b/src/psm_parser.py
@@ -150,10 +150,12 @@ def write_psm(name, items: list[str], filename: Path = Path("CodeServerConnector
         backup_filename = str(filename) + f".{datetime.now().strftime('%Y-%m-%d.%H%M%S')}.bak"
         shutil.copy2(filename, backup_filename)
 
-    output = get_all_code.replace("!NAME!", name)
+    output = "# ------- BEGIN CODE BLOCK -------\n"
+    output += get_all_code.replace("!NAME!", name)
     output += get_code.replace("!NAME!", name)
     output += convert_from.replace("!NAME!", name).replace("!MAPS!", define_maps(items))
     output += class_definition.replace("!NAME!", name).replace("!VARIABLES!", define_variables(items)).replace("!STRINGS!", define_strings(items))
+    output += "# ------- END CODE BLOCK -------\n"
     with open(filename, "a", encoding="utf-8-sig") as f:
         f.write(output)
     pass
\ No newline at end of file