improving setup.py