Create stage2.py