Added more tests to rspec file