Updated rake task