add grading functionality to test