Rake task modifications