Add better diffing to rake task