Problem 4 done