Little fix to Problem 2.