Update main notebook to for different grade A and C