Add solution to problem #7