Rake task modified