Python Homework 2