added Problem 4