Update main.py

second commit from Luming
Merge pull request #2 from noben/patch-2

Update main.py from luming