Add solution for problem 18