Update Homework 6.py