Update auto_grader.py