Python homework lab 6