Add rspec coverage