Intro to Python homework