Improving setup.py