Update main.py
Merge pull request #77 from danielcsims/master

Update main.py