Add rake task for testing