Add solution to problem #4