Tweak setup.py