diff --git a/mkdocs.yml b/mkdocs.yml
index 563a3b3d..98a7670a 100644
--- a/mkdocs.yml
+++ b/mkdocs.yml
@@ -129,6 +129,10 @@ nav:
               - QuickGELUActivation: "zeta/nn/modules/quickgeluactivation.md"
               - RecursiveBlock: "zeta/nn/modules/recursiveblock.md"
               - ReLUSquaredActivation: "zeta/nn/modules/relusquaredactivation.md"
+              - stochasticskipblock: "zeta/nn/modules/stochasticskipblock.md"
+              - gatedresidualblock: "zeta/nn/modules/gatedresidualblock.md"
+              - tripleskipblock: "zeta/nn/modules/tripleskipblock.md"
+              - DynamicRoutingBlock: "zeta/nn/modules/dynamicroutingblock.md"
           - zeta.nn.attention:
               - FlashAttention: "zeta/nn/attention/flash_attention.md"
               - MultiQueryAttention: "zeta/nn/attention/multiquery.md"
diff --git a/scripts/auto_tests_docs/mkdocs_handler.py b/scripts/auto_tests_docs/mkdocs_handler.py
index e25b2be5..9ded4215 100644
--- a/scripts/auto_tests_docs/mkdocs_handler.py
+++ b/scripts/auto_tests_docs/mkdocs_handler.py
@@ -22,8 +22,8 @@ def generate_file_list(directory, output_file):
                     # Remove the file extension
                     file_name, _ = os.path.splitext(file)
                     # Write the file name and path to the output file
-                    f.write(f'- {file_name}: "{directory}{file_path}"\n')
+                    f.write(f'- {file_name}: "{directory}/{file_path}"\n')
 
 
 # Use the function to generate the file list
-generate_file_list("docs/zeta/models", "file_list.txt")
+generate_file_list("docs/zeta/nn/modules", "file_list.txt")