Updated rspec to work with the code