lib/pry-singular.rb in pry-singular-0.1.1 vs lib/pry-singular.rb in pry-singular-0.1.2

@@ -2,23 +2,21 @@
 require 'pry'
 
 module PrySingular
   class << self
     def set_class(*klasses)
-      import_class_command(klasses)
+      klasses.each(&method(:import_class_command))
     end
 
     private
 
-    def import_class_command(klasses)
+    def import_class_command(klass)
       commands = Pry::CommandSet.new do
-        klasses.each do |klass|
-          klass.public_methods.each do |klass_method|
-            command "#{klass_method}", "#{klass}.#{klass_method}" do
-              klass.class_eval <<-EOS
-                #{Readline::HISTORY.to_a.last.gsub(' ', '')}
-              EOS
-            end
+        klass.public_methods.each do |klass_method|
+          command "#{klass_method}", "#{klass}.#{klass_method}" do
+            klass.class_eval <<-EOS
+              #{Readline::HISTORY.to_a.last.gsub(' ', '')}
+            EOS
           end
         end
       end
 
       Pry.config.commands.import(commands)
\ No newline at end of file