Add solution to problems 5 and 6