Update main.py

Main project code