Updated tests to pass in python 3