bin/ronin-gen in ronin-gen-0.1.0 vs bin/ronin-gen in ronin-gen-0.1.1

@@ -5,12 +5,15 @@
 lib_dir = File.expand_path(File.join(File.dirname(__FILE__),'..','lib'))
 unless $LOAD_PATH.include?(lib_dir)
   $LOAD_PATH << lib_dir
 end
 
+require 'ronin/ui/command_line/commands/gen'
 require 'ronin/ui/command_line'
 
-if ARGV.length > 0
-  ARGV[0] = "gen-#{ARGV.first}"
-end
+if (ARGV.length > 0 && ARGV.first[0..0] != '-')
+  ARGV[0] = "gen_#{ARGV.first}"
 
-Ronin::UI::CommandLine.run(*ARGV)
+  Ronin::UI::CommandLine.run(*ARGV)
+else
+  Ronin::UI::CommandLine::Commands::Gen.run(*ARGV)
+end