Changing import for Python3
Making is work in python2 as well