Add solution to problem #5