lib/gogo_driver.rb in gogo_driver-0.0.3 vs lib/gogo_driver.rb in gogo_driver-0.0.4

@@ -1,18 +1,17 @@
 require "gogo_driver/version"
 require "gogo_driver/entity"
 
-module GogoDriver
-  def go(url='https://www.google.com')
-    @entity = Entity.new unless @entity
+class GogoDriver
+  def initialize(url='https://www.google.com')
+    @entity = Entity.new
     @entity.go(url)
     @last_url = url
   end
-  module_function :go
 
   def method_missing(method, *args, &block)
     @entity.respond_to?(method) ? @entity.send(method, *args, &block) : super
   rescue Errno::ECONNREFUSED
-    go(@last_url)
+    initialize(@last_url)
     @entity.send(method, *args, &block)
   end
 end