Update main.py
Merge pull request #2 from simon-persson/patch-2

Update main.py