Update `setup.py` to install as Python package