Update main.py

Ongoing definition of the program structure.