Update main.py

Again another little edit before I have to go again.