Getting tests to pass in Python 3