Create main.py

flow defined