Create stage1.py