Added solution for problem 3