Add solution to first problem
Add solution to second problem