Python school work