Problem 4 and 5 solution