Python program to help grade assignments