Added solution to problem #6