Python HW4