Add rake task to test