Adding unittest support for python 3.x