Update main.py

Made everything great
Merge pull request #1 from zerfgog/master

Update main.py