Update main.py

quick path update