Update prob7.py