Updated rspec code