Add a rake task to run the tests