Update main.py
Merge pull request #4 from RoryBaxter/patch-3

Update main.py