Update main.py

first draft.