update Problem 4