Add solution to problem #3