Add main.py

This is the core of the program.