remove puts and inline variables.

2 files changed, 5 insertions, 5 deletions

diff --git a/lib/employee.rb b/lib/employee.rb
index b29c5be..794fc62 100644
--- a/lib/employee.rb
+++ b/lib/employee.rb
@@ -38,13 +38,9 @@ class Grant
 
   def vest_at(price, portion)
     number_of_units_to_vest = portion.to_f * @units.count
-    units_to_vest = @units.take(number_of_units_to_vest)
-    units_to_vest.each do |unit|
+    @units.take(number_of_units_to_vest).each do |unit|
       unit.vest_at(price)
     end
-    puts unvested_units.count
-    puts vested_units.count
-    puts @units.count
   end
 
   private
diff --git a/lib/unit.rb b/lib/unit.rb
index 0761c8e..efd73cb 100644
--- a/lib/unit.rb
+++ b/lib/unit.rb
@@ -1,4 +1,8 @@
 class Unit
+  def initialize
+    @vested = false
+  end
+
   def *(price)
     price
   end