Add solution for problem A2