Make writing grading code easier