adding Problem 4