Update main.py
Merge remote-tracking branch 'upstream/master'