Added solution for problem #6