Updated tests so that it runs with python 3