Update stages.py