diff --git a/src/main.py b/src/main.py
index f21d417..d6af54c 100644
--- a/src/main.py
+++ b/src/main.py
@@ -50,9 +50,7 @@ def get_updated_folders(path: str):
     
     return updated_folders
 
-def main():
-    site_path: str = argument_parsing()
-    folders = get_updated_folders(site_path)
+def gen_pages(folders: list[str]):
     with Bar("generating...", max=len(folders)) as bar:
         for folder in folders:
             files: list[str] = [os.path.join(folder, file) for file in os.listdir(folder)]
@@ -64,6 +62,11 @@ def main():
             readme: str = readmes[0] if len(readmes) > 0 else None
             bar.next()
 
+def main():
+    site_path: str = argument_parsing()
+    folders = get_updated_folders(site_path)
+    gen_pages(folders)
+
 
 if __name__ == "__main__":
     main()
\ No newline at end of file