Modify scenario.py