Intro to Python Class Homework