Resolved the compatibility issues of python2 and python3. The tests pass locally for both python 2 and python 3.