Added compatibility with python 3 in setup.py