python homework 1