Update Part B.py