Update Rakefile to run tests