Update main.py

more details in program structure added.