Updated tests to python 3
Changed to use relative imports