Update `setup.py`