Update main.py

corrected a few items.