Advance Python homework scripts