Update Step2.py