Update p21.py