Create main.py

Tested program.