Update pivot.py