refactor conditional using a hash.

1 files changed, 2 insertions, 9 deletions

diff --git a/lib/rover.rb b/lib/rover.rb
index 2f48cb5..f5bfa7d 100644
--- a/lib/rover.rb
+++ b/lib/rover.rb
@@ -59,15 +59,8 @@ class South
 end
 
 class Direction
+  @@directions = {:north => North.new, :east => East.new, :west => West.new, :south => South.new}
   def self.find(heading)
-    if(heading == :north)
-      return North.new
-    elsif heading == :east
-      return East.new
-    elsif heading == :south
-      return South.new
-    elsif heading == :west
-      return West.new
-    end
+    @@directions[heading]
   end
 end