Adding solution and rspec tests