Small additions to grading file