Got tests working for python 3