Added some rspec testing