Create main.py

The program.