Add missing tests for problem 11