Added solution for problem #3