Update main.py

minor fixes in names