Python script to automatically grade assignments