A2 Grade comments

Added the suggested modifications from the grading of A2