Add solution for problem A4