Added functionality to the grader to make it actually work