Adding code for grading assignment.