Add solution to problem #9