Solution problem 22
Add test specs to problem 22