changed imports so they work for python 2