Made the rake task simpler.