Update part1.py