Update tests to use rspec