Add unit test with rspec