Add solution to problem#2