Respond to PR feedback
Merge pull request #364 from edx/will/ai-grading-pr-feedback

Will/ai grading pr feedback