setup.py improved