setup.py improvements