Write some drafts in main.py