Add test spec to problem 20