added code for problem 1