Added code to highlight correct answer after test is complete