Update main.py

Python3 Support