Initial solution Problem 21
Add test spec to problem 21