Updated setup.py for pypi submission