Add tracking to rake tasks