update test_exam_mode to handle the new tests