Update pi.py