Corrected little mistakes in problem 4