Added solution for problem 1