Some sort of changes to main.py

Is this geezer dumb or something.