Update main.py

updated outline of the project