Update p004.py