Solution
Add test specs