Refactor sample data rake task