Create main.py

Main code file