Update 004.py