added code for problem 4