Update stage3.py