update problem 4