changed setup.py, so that it works with Python 2, too