Stack now implements Enuberable

pmatsinopoulos · Feb 12, 2012 · c2ea926 · c2ea926
1 parent c2e0776
commit c2ea926
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/models/stack.rb b/models/stack.rb
@@ -1,10 +1,13 @@
 class Stack
+  include Enumerable
 
   # public methods
   def initialize
     @stack = Array.new
   end
 
+  attr_reader :stack
+
   def push(element)
     @stack << element
   end
@@ -27,4 +30,8 @@ def empty?
     @stack.size == 0
   end
 
+  def each(&block)
+    @stack.each(&block)
+  end
+
 end