2 files changed, 20 insertions, 3 deletions
diff --git a/main.rb b/main.rb
index c04bd13..07b2fa9 100644
--- a/main.rb
+++ b/main.rb
@@ -13,9 +13,26 @@ class MazeGenerator
     @visitedCells = [@currentPos]
     @stack = [Pos.new(rand(@maze.width), rand(@maze.height))]
 
+    if ENV["DEBUG"] == "visual"
+      print "\e[?1049h" # Save the state of the terminal
+      print "\e[2J" # Clear the screen
+      print "\e[0;0H" # Move the cursor to 0, 0
+      print "\n   [ \e[1;35mGenerating maze...\e[0m ]\n\n" # Print some nice graphics
+      print "\e[s" # Save the cursor position
+    end
+
+    print "\e[s" if ENV["DEBUG"] == "visual"
+
     while !@stack.empty?
-      puts @maze if ENV["DEBUG"] == "visual"
       step()
+      if ENV["DEBUG"] == "visual"
+        print "\e[u" # Restore the cursor position
+        print @maze.to_s("   ")
+      end
+    end
+
+    if ENV["DEBUG"] == "visual"
+      print "\e[?1049l" # Restore the state of the terminal
     end
   end
 
diff --git a/maze.rb b/maze.rb
index d12a8f5..f3cbe1f 100644
--- a/maze.rb
+++ b/maze.rb
@@ -53,7 +53,7 @@ class Maze
     end
   end
 
-  def to_s
+  def to_s(prefix='')
     drawingField = Array.new(@height * 2 + 1) { " " * (@width * 2 + 1) }
 
     @v_walls.each_index do |x|
@@ -76,6 +76,6 @@ class Maze
       end
     end
 
-    drawingField.join("\n")
+    drawingField.map { |line| prefix + line }.join("\n")
   end
 end