lib/greenjaguar.rb in greenjaguar-0.0.4 vs lib/greenjaguar.rb in greenjaguar-0.0.5

@@ -6,20 +6,13 @@
 require 'greenjaguar/strategies/fixed_interval_strategy'
 require 'greenjaguar/policy_builder'
 require 'greenjaguar/retrier'
 
 module Greenjaguar
-
-  def self.included(klass)
-    klass.extend(ClassMethods)
+  def build_policy(&block)
+    PolicyBuilder.new(&block)
   end
 
-  module ClassMethods
-    def build_policy(&block)
-      PolicyBuilder.new(&block)
-    end
-
-    def robust_retry(policy, &block)
-      Retrier.new(policy, &block)
-    end
+  def robust_retry(policy, &block)
+    Retrier.new(policy, &block)
   end
 end
\ No newline at end of file