Added in regrading functionality

Also added automatic grading after training completes

Code review edits