Add rake integration