python homework