Little fix in problem 5.