Improve the grading function

[Hubert51]

The second answer sheet has different layout from first answer sheet. I want to use almost same code to grade different code. So some of code may need to be redesigned.