diff --git a/lib/pruner.rb b/lib/pruner.rb
index 015829b..e050aa0 100644
--- a/lib/pruner.rb
+++ b/lib/pruner.rb
@@ -15,6 +15,8 @@ module Pruning
 
       private
 
+      # Depth First Search implementation 
+      # With pruning on the same sweep 
       def prune_subtree(nodes, indicator_ids)
         nodes_to_examine = nodes.dup
         nodes_to_examine.each do |node|
diff --git a/test/test_helper.rb b/test/test_helper.rb
index 52f41a2..8a95ad6 100644
--- a/test/test_helper.rb
+++ b/test/test_helper.rb
@@ -5,7 +5,7 @@ require 'bundler/setup'
 require 'dotenv/load'
 require 'json'
 require 'minitest/autorun'
-require 'bogus/minitest/spec'
+require 'bogus/minitest'
 
 require_relative '../lib/api/app'