python homework 3