problem 2 modify