Update main.py

Main program outline.