improve p20.py