Get test_grade() to pass; unsure if correct