Update main.py

xrange is no longer supported in Python 3.x  Changed to range