Add rspec tests to problem 3