Clean up grading script