Accounted for some extra cases in the grading script.