Rakefile in runit-man-2.4.0a2 vs Rakefile in runit-man-2.4.0

@@ -1,8 +1,17 @@
 # encoding: utf-8
 
 require 'rubygems'
+
+begin
+  require 'bundler'
+  Bundler.setup(:default, :development, :test)
+  Bundler::GemHelper.install_tasks
+rescue LoadError
+  $stderr.puts "Bundler not available. Install it with: gem install bundler"
+end
+
 $LOAD_PATH.unshift('./lib')
 
 require 'runit-man/version'
 
 begin
@@ -21,12 +30,12 @@
 rescue LoadError
   $stderr.puts "RSpec not available. Install it with: gem install rspec-core rspec-expectations"
 end
 
 begin
-  require 'bundler'
-  Bundler.setup(:default)
-  Bundler::GemHelper.install_tasks
+  require 'yard'
+
+  YARD::Rake::YardocTask.new
 rescue LoadError
-  $stderr.puts "Bundler not available. Install it with: gem install bundler"
+  $stderr.puts "YARD not available. Install it with: gem install yard"
 end