fixing import for python 3