diff --git a/video_generator.py b/video_generator.py
index c2ff62b..f6e1659 100644
--- a/video_generator.py
+++ b/video_generator.py
@@ -1,5 +1,6 @@
 import os, shutil
 from moviepy.editor import concatenate_videoclips, VideoFileClip
+from utils.console import print_markdown
 from main import run
 
 class VideoGenerator():
@@ -21,22 +22,19 @@ class VideoGenerator():
             if os.path.exists("results"):
                 shutil.rmtree("results")
 
-        for _ in range(num_of_shorts):
-            print("="*50)
+        for i in range(num_of_shorts):
+            print_markdown(f"# Iteration {i}")
             run(self.config)
         
         print(f"Videos have been made!")
     
     def generate_long_video(self, num_of_videos_to_be_conc, remove_past_results=True):
-        if remove_past_results:
-            if os.path.exists("results"):
-                shutil.rmtree("results")
-            
+        if remove_past_results:            
             if os.path.exists("long_form_videos_results"):
                 shutil.rmtree("long_form_videos_results")
 
-        for _ in range(num_of_videos_to_be_conc):
-            print("="*50)
+        for i in range(num_of_videos_to_be_conc):
+            print_markdown(f"# Iteration {i}")
             run(self.config)
         
         print("Concatinating videos...")