updating stage.py