Update to rake task