HW6 Python