Add code to add grade