lib/eucalypt/migration/namespaces/migration-add/cli/add.rb in eucalypt-0.3.1 vs lib/eucalypt/migration/namespaces/migration-add/cli/add.rb in eucalypt-0.3.2

@@ -1,18 +1,16 @@
 require 'thor'
 require 'eucalypt/helpers'
 require_relative 'add-index'
 require_relative 'add-column'
+require 'eucalypt/list'
 
 module Eucalypt
   class MigrationAdd < Thor
     include Thor::Actions
+    extend Eucalypt::List
 
-    class << self
-      require 'eucalypt/list'
-      include Eucalypt::List
-      def banner(task, namespace = false, subcommand = true)
-        "#{basename} migration #{task.formatted_usage(self, true, subcommand).split(':').join(' ')}"
-      end
+    def self.banner(task, namespace = false, subcommand = true)
+      "#{basename} migration #{task.formatted_usage(self, true, subcommand).split(':').join(' ')}"
     end
   end
 end
\ No newline at end of file