Update p022.py