Enhance setup.py