Added assign_get_grades function