CODE: Make script works with python 2.6