bin/aipim in aipim-rails-0.0.179 vs bin/aipim in aipim-rails-0.0.181

@@ -2,18 +2,21 @@
 require 'aipim-rails'
 # root do aipim
 path = File.expand_path(File.dirname(__FILE__))+"/../"
 # pegar o nome dos arquivos das features
 def get_files
-	files = []
-	ls = %x[ls features].split("\n")
-	ls.each do |f|
-		f = f.split("/")
-		f.delete_at(0) if f[0] == "features"
-		f = f[0]
-		files << f if !(f =~ /.feature\z/).nil?
-	end
-	files
+        files = []
+        ARGV.delete_at(0)
+        ARGV.each do |arg|
+                ls = %x[ls #{arg}].split("\n")
+                ls.each do |f|
+                        f = f.split("/")
+                        f.delete_at(0) if f[0] == "features"
+                        f = f[0]
+                        files << f if !(f =~ /.feature\z/).nil?
+                end
+        end
+        files
 end
 
 if ARGV[0] == 'generate'
 	# Cria a pasta do aipim
 	system('mkdir -p aipim')
\ No newline at end of file