edited the rake task import