Add more rspec test coverage