use tap.

1 files changed, 5 insertions, 4 deletions

diff --git a/lib/gol/game.rb b/lib/gol/game.rb
index 7c26a87..81822da 100644
--- a/lib/gol/game.rb
+++ b/lib/gol/game.rb
@@ -4,10 +4,11 @@ class Game
   end
 
   def play(max_x, max_y)
-    world = World.new(create_cells(max_x, max_y))
-    until world.empty? do
-      @printer.display(world)
-      world.next_generation!
+    World.new(create_cells(max_x, max_y)).tap do |world|
+      until world.empty? do
+        @printer.display(world)
+        world.next_generation!
+      end
     end
   end