Week6 Python Assignment