make python script work in python 3