Update main.py
Merge pull request #71 from sasullivan/master

Update main.py