Added solution for problem 4