Create main.py

Added the only file of the program