improve setup.py