fixed some imports for python 2.x