Update stage-3.py