Updated setup.py for pypi distribution