Update main.py

a few updates...