Updated setup for PyPI