update Problem 2