Small fixes for the python grader