Update homework7.py