Update Files after Grading

Peer reviewer identified an improvement; this commit updated the files
to reflect the better approach after grading was complete.