diff --git a/lib/diffbench.rb b/lib/diffbench.rb
index a19eb4a..de8e26d 100644
--- a/lib/diffbench.rb
+++ b/lib/diffbench.rb
@@ -55,9 +55,9 @@ def run
         color_string = result_color(improvement)
         output "After patch:  #{after_patch.format}"
         output "Before patch: #{before_patch.format}"
-        #if color_string
-          #output self.class.color("Improvement: #{improvement}%", color_string).strip
-        #end
+        if color_string
+          output self.class.color("Improvement: #{improvement}%", color_string).strip
+        end
         output ""
       end
     end
diff --git a/spec/diffbench_spec.rb b/spec/diffbench_spec.rb
index 3020dc0..b20ef98 100644
--- a/spec/diffbench_spec.rb
+++ b/spec/diffbench_spec.rb
@@ -46,10 +46,12 @@ def to_regexp(output)
 --------------------------------------------------Sleeper 1
 After patch:    0.000000   0.000000   0.000000 (  0.100NUM)
 Before patch:   0.000000   0.000000   0.000000 (  0.200NUM)
+#{DiffBench::Runner.color("Improvement: 50%", :green)}
 
 --------------------------------------------------Sleeper 2
 After patch:    0.000000   0.000000   0.000000 (  0.100NUM)
 Before patch:   0.000000   0.000000   0.000000 (  0.200NUM)
+#{DiffBench::Runner.color("Improvement: 50%", :green)}
 OUT
     end
 
@@ -76,15 +78,13 @@ def to_regexp(output)
 --------------------------------------------------Sleeper 1
 After patch:    0.000000   0.000000   0.000000 (  0.100NUM)
 Before patch:   0.000000   0.000000   0.000000 (  0.200NUM)
+#{DiffBench::Runner.color("Improvement: 50%", :green)}
 
 --------------------------------------------------Sleeper 2
 After patch:    0.000000   0.000000   0.000000 (  0.100NUM)
 Before patch:   0.000000   0.000000   0.000000 (  0.200NUM)
+#{DiffBench::Runner.color("Improvement: 50%", :green)}
 OUT
-
-
-
-
       end
     end
   end