Changed the rake task