Update setup.py to be compatible with pypi