Fix the py file for Python 3+